diff --git a/Seg_UKAN/requirements.txt b/Seg_UKAN/requirements.txt
index 3c14a7c..ca31ca9 100644
--- a/Seg_UKAN/requirements.txt
+++ b/Seg_UKAN/requirements.txt
@@ -1,20 +1,17 @@
-addict==2.4.0
+addict
 dataclasses
 pandas
 pyyaml
 albumentations
 tqdm
 tensorboardX
-# mmcv-full==1.2.7
 numpy
 opencv-python
-perceptual==0.1
-pillow==8.4.0
-scikit-image==0.17.2
-scipy==1.5.4
-tifffile==2020.9.3
-timm==0.3.2
-typing-extensions==4.0.0
-yapf==0.31.0
-
-pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116
\ No newline at end of file
+perceptual
+pillow
+scikit-image
+scipy
+tifffile
+timm
+typing-extensions
+yapf
\ No newline at end of file
diff --git a/Seg_UKAN/train.py b/Seg_UKAN/train.py
index fd5ee00..0eb98dc 100644
--- a/Seg_UKAN/train.py
+++ b/Seg_UKAN/train.py
@@ -39,7 +39,7 @@
 from pdb import set_trace as st
 
 
-ARCH_NAMES = archs.__all__
+# ARCH_NAMES = archs.__all__
 LOSS_NAMES = losses.__all__
 LOSS_NAMES.append('BCEWithLogitsLoss')