Pytorch学习

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

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

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

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

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

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

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

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

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

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

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

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