Pytorch学习

2022-05-30马春杰杰评论(0)赞(0)

今天更换了一下卷积就这样了，还不清楚具体的原因，推测是pytorch多进程的问题。不过解决方案已经有了。 直接在主程序的开始加入下面这两句话即可： import multiprocessing as mp mp.set_start_meth...

2022-04-04马春杰杰评论(0)赞(1)

首先找到包含Trainer这个类的train.py文件，然后在这个类前面定义数据集： [crayon-651bb3a848b0d768059863/] 这里的__background__记得留着。 上面就把数据集定义好了，接下来要设置后面训...

2021-12-08马春杰杰评论(0)赞(3)

手动 举个例子： [crayon-651bb3a849960815511631/] 自动 a. 有序调整：等间隔调整(Step)，按需调整学习率(MultiStep)，指数衰减调整(Exponential)和 余弦退火CosineAnnea...

2021-07-30马春杰杰评论(0)赞(0)

简要概括就是：F.relu当函数用，nn.ReLU当模块定义用。 F.relu()一般在forward中用，nn.ReLU()在定义网络结构的时候用。 比如： [crayon-651bb3a849e3f253949396/] 注意：当用pr...

2021-05-22马春杰杰评论(0)赞(3)

看到一篇非常好的文章，介绍pytorch保存和加载模型的，非常详细。备份： 太长不看版： 方法1： torch.save(modelA.state_dict(), PATH) model.load_state_dict(torch.load...

2021-05-12马春杰杰评论(4)赞(2)

无法提供摘要。这是一篇受保护的文章。