xaddwell / Adversarial-Attack-on-Machinery-Fault-Diagnosis

The code for http://arxiv.org/abs/2110.02498
18 stars 2 forks source link

对于modelSave文件夹下的.pth模型没有和models的文件夹在的模型一一对应 #4

Open zy899 opened 1 year ago

zy899 commented 1 year ago

作者您好,很不好意思再一次打扰到您,为了进一步研习您上传的代码,发现了如题目所言的问题,models文件夹下的模型有NoBN的版本和BN版本,在modelSave下的模型只有一个版本,这个版本可以理解成是BN的还是NoBN的? 同时这个.pth模型在个人理解是需要训练后保存为.pth的,您的训练方法方便告知嘛? image

zy899 commented 1 year ago

补充:我是在报错如下图所示时,提出的这个疑惑 image