Training gets killed on 2nd epoch
sandeshshrestha45 opened this issue · 1 comments
sandeshshrestha45 commented
sadath-12 commented
@sandeshshrestha45 for testing it is set for default 2 . If you want to give your own epochs and train as you like you can remove --TESTING argument from python train.py --PARAMETERS ./parameters --BATCH_SIZE 2 --TIMESTEPS 25 --TESTING
to python train.py --PARAMETERS ./parameters --BATCH_SIZE 2 --TIMESTEPS 25 --EPOCHS 100