sshi commited on
Commit
e3b638c
·
1 Parent(s): bf27ec3

Update requirement file

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -9,6 +9,7 @@ PyYAML>=5.3.1
9
  requests>=2.23.0
10
  scipy>=1.4.1
11
  torch==1.7.0
 
12
  torchvision>=0.8.1,!=0.13.0
13
  tqdm>=4.41.0
14
  protobuf<4.21.3
 
9
  requests>=2.23.0
10
  scipy>=1.4.1
11
  torch==1.7.0
12
+ pytorch-lightning==1.9.3
13
  torchvision>=0.8.1,!=0.13.0
14
  tqdm>=4.41.0
15
  protobuf<4.21.3