Please provide the following information when requesting support.
• Hardware (T4/V100/Xavier/Nano/etc) A6000
• Network Type (Classification/pyt)
• Training spec file tao_tutorials/notebooks/tao_launcher_starter_kit/classification_pyt/classification.ipynb at main · NVIDIA/tao_tutorials · GitHub
• How to reproduce the issue ? (This is for errors. Please share the command line and the detailed log here.)
Having issue trying to train the example notebook. I have followed steps exactly. I have a conda env with python 3.8. I have also tested with an env with python 3.10
RandomResizedCrop.init() got an unexpected keyword argument ‘size’
Error executing job with overrides: [‘results_dir=/results/classification_experiment’, ‘model.init_cfg.checkpoint=/workspace/tao-experiments/pretrained_fan_hybrid_small/pretrained_fan_classification_imagenet_vfan_hybrid_small/fan_hybrid_small.pth’, ‘train.train_config.runner.max_epochs=3’]
/usr/local/lib/python3.10/dist-packages/nvidia_tao_pytorch/cv/classification/scripts/train.py FAILED
2 posts - 2 participants