Views
No views yet
stnr and install PyTorch. In the following installation steps, the default installed CUDA version is 12.1. If your CUDA version is not 12.1, please modify it according to the actual situation.1# Create conda environment
2conda create -n stnr python=3.8 -y
3conda activate stnr
4
5# Install PyTorch
6pip install -r requirements.txtpython main.py --file_root LEVIR --max_steps 80000 --model_type small --batch_size 16 --lr 2e-4 --gpu_id 0python eval.py --file_root LEVIR --max_steps 80000 --model_type small --batch_size 16 --lr 2e-4 --gpu_id 0├─Train
├─A jpg/png
├─B jpg/png
└─label jpg/png
├─Val
├─A
├─B
└─label
├─Test
├─A
├─B
└─label