Problems of package version

[stephaniekuo0201]

May I ask what version do you use for torch, torch_vision and Cuda? It seems to have some problems running your code on colab because of the version.

[g1910]

Hi @stephaniekuo0201, the torch and torchvision versions used are rather old given when the code was initially developed. Most likely <1.0. If you have some specific issue, you can paste your error here and I can help debug. Thanks!