Description of changes: In the bash scripts, we use the command line flag as "per_gpu_batch_size" but in the python script, we have used "batch_size". So, fixing it.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.
Issue #, if available:
Description of changes: In the bash scripts, we use the command line flag as "per_gpu_batch_size" but in the python script, we have used "batch_size". So, fixing it.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.