I have started to train SAN but I have some mistake that I could not understand in line:

start train itn-cpm model

itn_cpm = models.dict[args.arch](args.modelconfig, args.cycle_model_path)
procedure.train_san_epoch(args, itn_cpm, train_loader, eval_loaders, log)
Error:
AttributeError: 'ITN_CPM' object has no attribute 'module'
Can you help me with this error?