D-X-Y / landmark-detection

Four landmark detection algorithms, implemented in PyTorch.

https://xuanyidong.com/assets/projects/TPAMI-2020-SRT.html

MIT License

925 stars 180 forks source link

SAN Training error #69

Open UmirT opened 4 years ago

UmirT commented 4 years ago

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?

sevaraMS commented 4 years ago

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 4 years ago

Cool~

D-X-Y commented 4 years ago

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