/FSL-TPN

reproduce TPN

Primary LanguagePython

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
conda install -c conda-forge python-lmdb
pip install opencv-python tqdm pyarrow tensorboard
CUDA_VISIBLE_DEVICES=0 python train.py