Closed nianshaoyouweibuzibei closed 2 years ago
By setting the value of --bs. For example, the command for training with a batch size of 24 is python train_net.py sad --bs 24; the command for training with a batch size of 16 is python train_net.py sad --bs 16.
python train_net.py sad --bs 24
python train_net.py sad --bs 16
By setting the value of --bs. For example, the command for training with a batch size of 24 is
python train_net.py sad --bs 24; the command for training with a batch size of 16 ispython train_net.py sad --bs 16.