AutoML for Classification with PyTorch?

Please provide the following information when requesting support.

• Hardware (T4/V100/Xavier/Nano/etc)
• Network Type (Detectnet_v2/Faster_rcnn/Yolo_v4/LPRnet/Mask_rcnn/Classification/etc) Classification_pyt
• TLT Version (Please run “tlt info --verbose” and share “docker_tag” here) v5.0.0
• 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.)

Based on the Github file here, I see that every single parameter has been marked disbled for auto ML, so we cannot use AutoML with Pytorch?

Unfortunately yes, from AutoML - NVIDIA Docs, if the automl_enabled column is set to neither True nor False, the corresponding parameter can be added into additional_automl_parameters by user.
The workaround will still be building a custom docker. We will improve it in next release.

1 Like

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.