hw2hwei

I solved this problem by updating the Pytorch version.

> 您好： > 请问为什么我运行build_list.py后，得到的train_split.txt和val_split.txt都是空的。 > > 期待您的解答！感谢您百忙之中抽空回复！ > 李蔚 你好，关于数据集储存格式已经在ReadMe更新了，如果后续还有问题的话，请发邮箱[email protected]。

> 您好，我将baseline用resnet50后，训练S2时总是会突然停顿下来不再训练！ 请问是什么原因造成的？想请教一下您？ 万分感谢！ 我自己没用过resnet50，并没有遇到过这个问题，具体描述一下是怎么停顿下来的？

> 就是训练s2时，也不会报错就是直接停在某个epoch不动了 如果是用笔记本跑的话，可能是内存不够了，换resnet34对比试试吧

> 您好，怎么训练两个呀？s1和s2 Usage里面有介绍怎么训练。 以AID，resnet18为例，先运行python main.py --dataset AID --arch resnet18 --mode s1，再运行python main.py --dataset AID --arch resnet18 --mode s2

> 连续执行两次main文件吗？ 是的，等s1训练完后再运行s2

> 您好，为什么test文件当执行s1时就会报错呀？ 显示： heat_map = heat_map[0].cpu().detach().numpy() TypeError: 'NoneType' object is not subscriptable 你需要一步步地debug一下，看看是哪一步的数据生成错了、或者没有生成/读取到

> > 设置为s2时正常运行，设置s1时就报错了 当前代码在设置s2训练时，会把s1的参数覆盖掉，如果你想保留s1的参数并进行测试的话需要修改一下代码

> 打扰您了，我还有一个疑问。训练完s1和s2后分别得到了s1和s2的准确度，要自己手动计算两个准确度的平均值么？要怎么发现最终的准确度呢？ 源代码里面有，当mode==s2的时候进行求s1、s2的平均值。