Bad very long time to train compare the original project DmitryUlyanov

[putuoka]

I run !python train.py -content input/boy18.wav -style input/girl52.wav on google colab and it tooks long time to finish

torch.Size([1, 1, 257, 244])
torch.Size([1, 1, 257, 355])
1000 5.0% 1m 38s content_loss:3.222046 style_loss:183.106659 total_loss:186.328705
2000 10.0% 3m 19s content_loss:3.329478 style_loss:32.088646 total_loss:35.418125
3000 15.0% 4m 59s content_loss:3.537519 style_loss:12.308197 total_loss:15.845716
4000 20.0% 6m 40s content_loss:3.672033 style_loss:4.532965 total_loss:8.204998

and still keep going

in original project by dimitry this doesn't happen