AttributeError: 'augment_inputs_network' object has no attribute 'module'. Did you mean: 'modules'?

[hepburnalex]

训练的时候报了这样的一个错误：“ File "/DeepFakeDefenders/main_train_single_gpu.py", line 130, in
train_engine.save_checkpoint(log_root, epoch_idx, train_top1, val_top1, ema_start)
File "/DeepFakeDefenders/core/mengine.py", line 243, in save_checkpoint
ema_stact_dict = self.ema_model.module.module.state_dict()
File "/root/miniconda3/envs/deepfake/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1269, in getattr
raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'augment_inputs_network' object has no attribute 'module'. Did you mean: 'modules'?”
请问您遇到过吗？

[Fieldhunter]

您好，这里有两个module，其中的一个module本来是用于多卡分布式模型的，在单卡的时候应该去掉一个module。这里并没有针对单卡训练做修改，后续会针对这里的问题进行修复。很抱歉对您在使用我们代码的过程中造成困扰，您可以去掉一个module再尝试一下。如有其他问题，欢迎随时提出issue。