search

MASILab / 3DUX-Net

244 stars 34 forks source link

An error occurred during training #31

Closed cyuan031 closed 1 year ago

cyuan031 commented 1 year ago

Hi , thank you very much for your great work，I have meet some trouble when I train the amos dataset，it just shows ： 1683630496409 I wander why it happened，how to solove it

leeh43 commented 1 year ago

Hi, thank you for your interest towards our work! Please change args.pretrain_classes with int(args.pretrain_classes) and it should be fine. Current error is referring that the args.pretrain_classes is in string format, not in int format for input.

cyuan031 commented 1 year ago

Thank you for your answer. The problem has been resolved