D-X-Y/landmark-detection

SAN Training error

UmirT opened this issue · 3 comments

UmirT commented

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?

Sorry it was my mistake. In training process i used .DET, but when we train with our dataset we have to use .GTB

D-X-Y commented

Cool~

D-X-Y commented

@UmirT Did you still have this issue with the latest code?