Saving model in pytorch

torch.save(model_ft, "modelFT_60.pth")

댓글

이 블로그의 인기 게시물