How to change the model in tutroial - "Re-training on the Cat/Dog Dataset" by using "--arch"?

Hello, im doing tutorial by Dusty called “Re-training on the Cat/Dog Dataset”. This Tutorial is using the ResNet-18 model . I want to change to a different model. To change model, it says - " you can change that with the --arch flag." So my question is how to use --arch to change model , i tried this, but not working - “python3 train.py --model-dir=models/cat_dog --batch-size=4 --workers=1 --epochs=1 –arch=AlexNet data/cat_dog”

here is link tutorial :

Hi @betman15050, can you try it with --arch=alexnet? Thanks.

Hello Dusty, sorry for taking your time, yes it works that way. Thank you.

вт, 29 нояб. 2022 г. в 23:23, dusty_nv via NVIDIA Developer Forums <notifications@nvidia.discoursemail.com>:

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