Code will stop when running on the line "total_loss.backward()"

mysee1989 / TCAE

Self-supervised Representation Learning from Videos for Facial Action Unit Detection

162 stars 27 forks source link

Code will stop when running on the line "total_loss.backward()" #6

Closed yy19931029 closed 4 years ago

yy19931029 commented 4 years ago

When I run your code to train, it stop at the line 314 "total_loss.backward()" in "self_supervised_train_TCAE.py", it result no error and warning so I don't know how to continue my training.

yy19931029 commented 4 years ago

I use python 3.5 to run your code~

yy19931029 commented 4 years ago

I done! My gpu memory is not enough