diff --git a/requirements.txt b/requirements.txt index d8b40b456af0bdef955f83d5b82ec65113c6d4ac..fb1b543e039eb6f5b4f299be59b010161258dc8a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ pytorch-lightning >= 0.9.0 -torch -torchvision \ No newline at end of file +torch >= 1.3.0 +torchvision >= 0.6.0