I have build jetson-inference and trying to retrain cat-dog model.
When I run the below command, it starts training and I can see the first epoch completed print on the screen. After that, in about a minute the jetson nano unit shutdowns.
python3.6 train.py --model-dir=cat_dog ~/datasets/cat_dog
Pytorch version: 1.3.0
Jetson sdk: 4.3
This issue happening always. I need to restart the unit. If I set the epoch’s default count to 1 instead of 35, then I can see this completely running and generating the model. So if the train.py runs for about more than a minute the system shutdowns.
Let me know what is wrong here and the solution for this. Thank you.