Thomas.Chaigneau commited on
Commit
60b60e4
1 Parent(s): ce75d3f

add requirements

Browse files
Files changed (1) hide show
  1. requirement.txt +3 -0
requirement.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ torch
2
+ torchvision
3
+ pytorch-lightning