M2WindRunner

如何调整输入图片的尺寸

13

我用常用的参数训练发现超过了显存，其他的stanet代码里有 --load_size 和 --crop_size 的参数，在调小 load_size 参数（输入网络的参数）之后即可训练，请问您的代码里有没有类似的参数可以调节呢？

某个模块的实现方法

1

您好，请问您是如何实现取回原数据的文件名的呢？（即将图片以tensor格式载入之后，如何再获取原来的文件名，或者请告知eval部分的实现代码在哪？）

AssertionError: The directory exists, modify `model_name` in train.json 确定model_name一栏写的resnet