torch.nn.modules.module.ModuleAttributeError: 'TaxiBJTrainer' object has no attribute 'seek'

[qiaoyifan]

The error "torch.nn.modules.module.ModuleAttributeError: 'TaxiBJTrainer' object has no attribute 'seek'" occurs when I set the resume=True. I change the code in trainer.py
if resume:
checkpoint = torch.load(self, ckpt_path)
epoch = checkpoint["epoch"]
model.load_state_dict(checkpoint["state_dict"])
to
if resume:
checkpoint = torch.load(ckpt_path)
epoch = checkpoint["epoch"]
self.load_state_dict(checkpoint["state_dict"])
which fix the error.

[zhaojieting]

Thanks, help a lot.