Please provide the following information when requesting support.
• Hardware : GTX 3090
• Network Type : Classification
• TLT Version (Please run “tlt info --verbose” and share “docker_tag” here): TAO API
• Training spec file(If have, please share here)
• How to reproduce the issue ? (This is for errors. Please share the command line and the detailed log here.)
Hello,
I am using the client notbook for AUTOML on classification (multiclass), When I want to use the effientnet B0 or B1, I have problems. Resnets and darknet seems to be working.
On EfficinetnetB1, The trainng stuck at epoch zero and dont show any error message.
On EfficinetnetB0, it can not find any pretrain model of classification models.
Could you help with this?
yes, download link is same. the file is in folder: 'getting_started_v4.0.0/notebooks/tao_api_starter_kit/client/automl"
I suspect there some modification on the train config file thatshould be done (number of lyers or activation). did you be able to run with B1?
For B0, it is looking for ptm item number in the notebook, where would I put the link from your suggestion in the notebook?