Open YevheniiSemendiak opened 2 years ago
The train.py code in the repo doesn't work with PyTorch higher than 1.4.0 due to API changes. This PR fixes PyTorch to be in between 0.4.0 and 1.4.0 versions.
train.py
The
train.py
code in the repo doesn't work with PyTorch higher than 1.4.0 due to API changes. This PR fixes PyTorch to be in between 0.4.0 and 1.4.0 versions.