真的有人能跑起来吗？

[begin20231024]

core/mengine.py

这个文件的坑有很多，TrainEngine类甚至都没有self.ema_model，而代码中确调用了self.ema_model.module和self.ema_model.module.module

跑起来的时候报错：
Traceback (most recent call last): File "/Defenders/main_train_single_gpu.py", line 130, in
train_engine.save_checkpoint(log_root, epoch_idx, train_top1, val_top1, ema_start)
File "Defenders/core/mengine.py", line 244, in save_checkpoint
ema_stact_dict = self.ema_model.module.module.state_dict()
File "/home/anaconda3/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'?

花了很多时间跑这个代码，希望作者能够解答

[Drahann]

#13 (comment)
我是把self.ema_model.module.module中删掉一个module