fix main to take into consideration value of batches_per_epoch argument +...
fix main to take into consideration value of batches_per_epoch argument + removed batches_per_epoch from template since it is not used
Please register or sign in to comment