Training stuck at epoch 1

[Suppersine]

I was deploying your model on COLAB - with the repo copied, I ran the code below:

!python main.py --mode train \ --attention_len 16 \ --batch_size 32 \ --data_set muse \ --dropout 0.2 \ --learning_rate 1e-5 \ --model_dir ./models/model \ --num_epochs 40 \ --num_layers 3 \ --num_units 338

Everything seems to be fine at that point, but it's stuck while training - I don't know why.T_T https://colab.research.google.com/drive/1jaHTWg637wkD35C5LC5rRjej_G95L6tV?usp=sharing

[liuyue777]

I was deploying your model on COLAB - with the repo copied, I ran the code below:

!python main.py --mode train --attention_len 16 --batch_size 32 --data_set muse --dropout 0.2 --learning_rate 1e-5 --model_dir ./models/model --num_epochs 40 --num_layers 3 --num_units 338

Everything seems to be fine at that point, but it's stuck while training - I don't know why.T_T https://colab.research.google.com/drive/1jaHTWg637wkD35C5LC5rRjej_G95L6tV?usp=sharing

Hi, I have the same problem, have you found a solution?