https://github.com/lukemelas/EfficientNet-PyTorch/issues/351 #352

Activity

1. train-py commented on Jul 17, 2025

   

   train-py

   on Jul 17, 2025

   Author

   More actions

   Ok thanks
2. train-py commented on Jul 17, 2025

   

   train-py

   on Jul 17, 2025

   Author

   More actions

   https://github.com/lukemelas/EfficientNet-PyTorch/issues/351torchrun --nproc_per_node=8 train.py
   --model $MODEL --epochs 600 --opt rmsprop --batch-size 128 --lr 0.064\
   --wd 0.00001 --lr-step-size 2 --lr-gamma 0.973 --auto-augment imagenet --random-erase 0.2torchrun --nproc_per_node=8 train.py
   --model mobilenet_v2 --epochs 300 --lr 0.045 --wd 0.00004
   --lr-step-size 1 --lr-gamma 0.98torchrun --nproc_per_node=8 train.py
   --model $MODEL --epochs 100torchrun --nproc_per_node=8 train.py --model inception_v3
   --test-only --weights Inception_V3_Weights.IMAGENET1K_V1100