training stops in colab

[ghost]

I am using these parameters python train.py --batch_size 4 --exp TEST --frame_sizes 16 4 --n_rnn 2 --dataset piano
and training stops in colab with ^C automatically

[michaelybecker]

I'm able to train a bespoke model on colab just fine. Training continues until stopped or until the Colab hits the 12 hour limit (when it expires). Samples get generated every once in a while (end of epoch, I think). Can you elaborate on your problem?

[ghost]

@MichaelHazani can you share your colab notebook?

[michaelybecker]

@ak9250 Sure, here's a reference Colab notebook I just made:
https://colab.research.google.com/drive/1tc2fzngIsnRt-ut21W7ADlU3yUzUqtw0
(only ran the RNN for a few seconds to make sure it's working)

[phatmann]

@MichaelHazani do you have an updated version of the notebook?