qic999's picture
Upload folder using huggingface_hub
28e6f98 verified
|
Raw
History Blame Contribute Delete
427 Bytes

Dependency

The code is tested on python 3.8, Pytorch 1.13.

Setup environment
conda create -n FSMNet python=3.8
source activate FSMNet # or conda activate FSMNet
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
pip install einops h5py matplotlib scikit_image tensorboardX yacs pandas opencv-python timm ml_collections