博文

Materials about Pytorch installation on Windows

2018-11-13 15:55

1. 安装Anaconda 本人安装3.7 2.安装cuda时，参考 https://www.youtube.com/watch?v=HExRhnO5Mqs 注意，在安装cuda之后，若询问是否安装visual studio code，选择是。因为visualstudiocode是启动显卡驱动需要的库。也可以参考上述视频的方法安装visualstudio。 ...

个人分类: DeepLearning|1456 次阅读|没有评论

DL--Examples of weight Initialization in deep neural network

2017-12-13 07:26

I. Example in DCGan: https://github.com/pytorch/examples/blob/master/dcgan/main.py Step 1. Set seed if opt.manualSeed is None : opt.manualSeed = random.randint( 1 , 10000 ) print ( Random Seed: , opt.manualSeed) random.seed(opt.manualSee ...

个人分类: DeepLearning|4300 次阅读|没有评论

Pytorch--save best model in the middle of training process

2017-11-9 01:02

I'd like to save the best_model during training process with epoch_num epochs. I tried several ways, but only the way of saving the middle best_model to local computer works. The deepcopy (self.best_model = copy.deepcopy(self.model_ft)) did not work The model assignment (self.best_mode ...

个人分类: DeepLearning|2891 次阅读|没有评论

Pytorch--Bug--tensors are on different GPUs

2017-11-4 07:42

Traceback (most recent call last): File cluster_resnet_9.py, line 361, in module train_model.save_model(model_path) File cluster_resnet_9.py, line 211, in eval_model if self.cuda_id -1: File /home/linlin/anaconda2/lib/python2 ...

个人分类: DeepLearning|5582 次阅读|没有评论

本页有 6 篇博文因作者的隐私设置或未通过审核而隐藏