tiangexiang / SQUID

[CVPR 2023] Deep Feature In-painting for Unsupervised Anomaly Detection in X-ray Images
MIT License
90 stars 6 forks source link

关于缺少in_channels参数 #13

Open Codierungskatz opened 2 months ago

Codierungskatz commented 2 months ago

几位老师你们好,我在一门课上选择了这篇文章来尝试复现代码,在此过程中遇到一个问题。 当我想运行digit_best.py时,得到了以下错误提示:Traceback (most recent call last): File "main.py", line 43, in model = AE(1, 32, CONFIG.shrink_thres, num_slots=CONFIG.num_slots, num_patch=CONFIG.num_patch, level=CONFIG.level, File "C:\Users\Erwin\Desktop\SQUID\SQUID-main\models\squid.py", line 53, in init self.memory_blocks.append(getattr(Memory, config['type'])(num_slots=mem_num_slots, slot_dim=self.filter_list[i] * config['multiplier'], TypeError: init() missing 1 required positional argument: 'in_channels' 对于200个epochs,我自行添加了几个in_channels赋值来进行尝试,但始终没有得到比较好的结果。请问对于此变量,是否可以给我些建议,比如in_channels=self.filter_list[i]或者in_channels=num_in_ch 希望能得到您的回复,谢谢!