Commit 5b4b8d31 authored by 李聪聪's avatar 李聪聪
Browse files

更新 README.md

parent 0c91dd65
Loading
Loading
Loading
Loading
+26 −0
Original line number Diff line number Diff line
@@ -8,3 +8,29 @@ Unsupervised domain adaptation is critical in various computer vision tasks, suc

![SAPNET](SAPNET.png)

## Install 

Requirements (Please install latest version):

```
torch
torchvision
terminaltables
Pillow
yacs
pycocotools
```

## Run

Train a model:

```
python -m torch.distributed.launch --nproc_per_node=8 --master_port=10086 --use_env adversarial_train.py --config-file <path_to_config>
```

Test a model:

```
python -m torch.distributed.launch --nproc_per_node=8 --master_port=10086 --use_env adversarial_train.py --config-file <path_to_config> --test-only
```