update file: requirment.txt

DevoLearn / devolearn

Accelerate data driven research on embryos with deep learning models :microscope:

MIT License

43 stars 30 forks source link

Closed vrutikrabadia closed 3 years ago

vrutikrabadia commented 3 years ago

Updated dependencies for PyTorch and torchvision.

Resolve #42

Mayukhdeb commented 3 years ago

Hi @vrutikrabadia, thanks for your PR, can we make it work with both versions i.e 1.7.0 and 1.8.0 ? Something like:

torch>=1.7.0
torchvision>=0.8.1

vrutikrabadia commented 3 years ago

Thanks I'll do the required

vrutikrabadia commented 3 years ago

Hello, @Mayukhdeb, I have added the changes you asked for.