1. Main.py
60 lines:
parser.add_argument('--lambda_id',type=float,default=10,help='weight for identity loss')
96 lines:
parser.add_argument('--rafd_image_dir',type=str,default='data/RaFD/train')
2. lines = [line.rstrip()for linein open(self.attr_path, 'r')]
(1)Python strip() 方法用于移除字符串头尾指定的字符(默认为空格或换行符)或字符序列。
(2)Python rstrip() 删除 string 字符串末尾的指定字符(默认为空格).
其实对于STARGAN说明的博客有很多:
先放在这里,稍后补充(因为code 都没跑通,没资格分析,orz
——————————————————————————分割线 2019/11/12____________________________
STARGAN的测试问题:
(1)调用python main.py 函数的时候,如何把参数输进去
这里感谢王同学的面授...
(2)FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecate
错误提示如:
报错原因是因为tensorflow numpy的问题,可以选择更换numpy版本 (pip install numpy==1.16.0),或者直接按照错误提示,将1,修改为(1,),不要忘记加括号,不然还要再进一次!!
参考:【TensorFlow】FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecate解决方法
链接:https://blog.csdn.net/Vici__/article/details/100903453
参考:解决python调用TensorFlow时出现FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecate
链接:https://blog.csdn.net/bigdream123/article/details/99467316
(3)提示 no such file or directory!!
但是明明是有的,那就先替换,结果替换了这一个,其他的又找不到了,感觉像是随机的。
这他喵的竟然真的是 因为我没有建立images!!
数据问题!!搞了好几次,每次都重新上传pycharm(先直接放data,又直接放celeba下,明明能看懂code 但智障小D真的无语)
建立之后就可以start trainging了
(4)RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 2.00 GiB total capacity; 1.15 GiB already allocated; 103.17 MiB free; 46.92 MiB cached)
因此:
无效。。先去吃午饭
———————分割线——————
下午请师兄来帮忙,可是师兄不熟悉win系统,还连连让我装上双系统,和服务器 (这不是意味着我这周工作白做了...
晚上吃饭又仔细看了一下GPU, 这个GTX960只有2G显存,刚开始train model的时候就爆了,直接崩了,不知道咋整
于是又麻烦了王同学来帮忙
关于如何查看GPU占存的方法可以参考:
【windows查看GPU占用率以及GPU未执行原因】
https://blog.csdn.net/weixin_41770169/article/details/80349088
——————分割线留给王同学—————————————
王同学来过了
这么一交流,1080 8GB内存,跑起来毫无压力
所以 我的960 2G 弱爆....
就借用王同学的电脑跑跑看效果了
这里还发现了贴吧 2080 11G的 train 18h 的留言
溜了溜了
所以很有可能王同学的test也不行....
静候佳音了,另外,王同学还说
跑这种代码,要先找找有没有训练好的model,直接拿过来看看效果
闲了再看看 github的另一个源码 https://github.com/taki0112/StarGAN-Tensorflow (原始模型跑通了,用不上了哈哈)
————————2019/11/23 测试成功分割线———————
话题还是和王同学有关的
帮我搞定了服务器的配置,这么多工作记录可以想到我的配置之路是多么的坎坷了
王同学真的是厉害又耐心
就这样,人生中的第一个模型测试成功。
偷偷放上一张和王同学的聊天记录
天!开心