train_CNN.py文件

[qujinjie]

作者你好，请问train_CNN.py文件中model.load_state_dict(torch.load("./checkpoints/cnn_RGBdiffer_model0.9958.pth"))的cnn_RGBdiffer_model0.9958.pth在那下载

[GuardSkill]

你好，可以注释这步（这一步其实是用来接着之前的断点继续训练的），自己训练CNN模型，这个模型就是用来训练论文中所说的CNN，但是他们用的那个wild 数据集。