The training script, train.py only accepts the number of epochs and learning rate. The default batch size is set t to 1 and as a result the training has been very slow. It will be good to be able handle more arguments and see how they affect performance.
The training script,
train.pyonly accepts the number of epochs and learning rate. The default batch size is set t to 1 and as a result the training has been very slow. It will be good to be able handle more arguments and see how they affect performance.Acceptance criteria: