AssemblyAI-Community/MinImagen

Training gets killed on 2nd epoch

sandeshshrestha45 opened this issue · 1 comments

I've tried training with this command: python train.py --PARAMETERS ./parameters --BATCH_SIZE 2 --TIMESTEPS 25 --TESTING

but the training gets killed at the second epoch.
image

@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