Code 测试(1):StarGAN model

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 字符串末尾的指定字符(默认为空格).


restrip和strip的区别

其实对于STARGAN说明的博客有很多:

先放在这里,稍后补充(因为code 都没跑通,没资格分析,orz

——————————————————————————分割线 2019/11/12____________________________

STARGAN的测试问题:

(1)调用python main.py 函数的时候,如何把参数输进去

这里感谢王同学的面授...


这里注意把github给出来的 \去掉,不然会报错,其他的复制就可以了

(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)

说是batchsize太大的原因

因此:


main.py 从20修为10


data_loader.py 从16修为4

无效。。先去吃午饭

———————分割线——————

下午请师兄来帮忙,可是师兄不熟悉win系统,还连连让我装上双系统,和服务器 (这不是意味着我这周工作白做了...

晚上吃饭又仔细看了一下GPU, 这个GTX960只有2G显存,刚开始train model的时候就爆了,直接崩了,不知道咋整

于是又麻烦了王同学来帮忙

关于如何查看GPU占存的方法可以参考:

【windows查看GPU占用率以及GPU未执行原因】

https://blog.csdn.net/weixin_41770169/article/details/80349088


GPUZ检测到 gpu load即将爆

——————分割线留给王同学—————————————

王同学来过了

这么一交流,1080 8GB内存,跑起来毫无压力

所以  我的960 2G 弱爆....

就借用王同学的电脑跑跑看效果了

这里还发现了贴吧 2080 11G的 train 18h 的留言

溜了溜了

所以很有可能王同学的test也不行....

静候佳音了,另外,王同学还说

跑这种代码,要先找找有没有训练好的model,直接拿过来看看效果

闲了再看看 github的另一个源码  https://github.com/taki0112/StarGAN-Tensorflow (原始模型跑通了,用不上了哈哈)

————————2019/11/23  测试成功分割线———————

话题还是和王同学有关的

帮我搞定了服务器的配置,这么多工作记录可以想到我的配置之路是多么的坎坷了

王同学真的是厉害又耐心

就这样,人生中的第一个模型测试成功。

偷偷放上一张和王同学的聊天记录

天!开心

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容