02 2021 档案

摘要:训练使用的是天池的GPU,没有用到tensorBroadX,用的是静态的plt显示Loss。 我通过实验证明了同样的网络使用WGAN-GP的网络架构对比没有使用WGAN-GP的网络架构会有更加不容易模型崩溃(model collapse)。 下面是代码,使用的数据集CIFAR10。 utils.py 阅读全文
posted @ 2021-02-22 16:49 安智伟 阅读(408) 评论(0) 推荐(0)
摘要:模型用的是苹果转橘子的数据集。但可能是由于模型太大且图片数量不足(1000张左右)。因此,有些图片transform不是很好。 模型是挂在天池上面跑的。还需要导入until.py文件,我放在文末了。 import glob import random import os import torch f 阅读全文
posted @ 2021-02-08 19:24 安智伟 阅读(666) 评论(0) 推荐(0)
摘要:RuntimeError: The size of tensor a (40) must match the size of tensor b (41) at non-singleton dimension 3 此类还是维度问题,建议尝试解决方案: 把最后一轮训练跳过就好了 阅读全文
posted @ 2021-02-06 21:16 安智伟 阅读(16885) 评论(0) 推荐(0)