请问训练参数问题

[hugefrog]

请问，论文里给的训练参数和代码里给的训练参数好像优点差别，以哪个参数为准呀？

[WestdoorSad]

是哪个参数有出入呀，以代码为准吧

[hugefrog]

是下面几个参数哦，麻烦了。
1.用HisarMod2019.1数据集的时候，dataset.py中stft的window_size需要改成论文中的61吗？生成以后截取的频谱宽度还是保持32吗（stp[:32,:]）？
,,stp = stft(self.samples[idx][0,:],1.0,'blackman',31,30,128)
return torch.Tensor(self.samples[idx]),torch.Tensor(np.expand_dims(stp[:32,:],0)), self.SNR[idx], self.label[idx]

2. main.py里面patience=10参数是论文里的Early stop=20的参数吧？

3.main.py里面构建IQFormer时候传的layers（[2,3,2]和[3,3,3]）参数就是论文中的Number of Conv Encoder([1,2,1],[2,2,2])参数吗？
if args.database_choose in ['2016.10a','2016.10b']:
model = IQFormer([2,3,2], embed_dims=[64,64,64],
mlp_ratios=4,
act_layer=nn.GELU,
num_classes=num_classes,
down_patch_size=3, down_stride=2, down_pad=1,
drop_rate=0.2, drop_path_rate=0.,
use_layer_scale=False, layer_scale_init_value=1e-5,
fork_feat=False,
vit_num=1,)
else:
model = IQFormer([3,3,3], embed_dims=[64,64,64],
mlp_ratios=4,
act_layer=nn.GELU,
num_classes=num_classes,
down_patch_size=3, down_stride=2, down_pad=1,
drop_rate=0.2, drop_path_rate=0.2,
use_layer_scale=False, layer_scale_init_value=1e-5,
fork_feat=False,
vit_num=1,)

[WestdoorSad]

哦哦 1要改成61 2早停机制是20没错 3的话就是你理解的那样

[hugefrog]

好的感谢，我去试一下。