GuHuangAI
diff --git a/‎.idea/workspace.xml
+9-5 b/‎.idea/workspace.xml
+9-5
diff --git a/‎requirement.txt
+3-3 b/‎requirement.txt
+3-3
@@ -1,12 +1,12 @@
 #torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
-accelerate
+accelerate==0.16.0
 einops
 ema-pytorch
-pytorch-lightning
+pytorch-lightning==1.9.3
 scikit-learn
 scipy
 thop
-timm
+timm==0.6.12
 tensorboard
 fvcore
 albumentations