Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
added command for single and multi-temporal regression finetuning
  • Loading branch information
RituYadav92 authored Sep 20, 2024
1 parent 4f598fe commit 9fe0f05
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -130,12 +130,24 @@ torchrun --nnodes=1 --nproc_per_node=1 run.py \
```
#### Single Temporal Regression
```
torchrun ...
torchrun --nnodes=1 --nproc_per_node=1 run.py \
--config configs/run/default.yaml \
--encoder_config configs/foundation_models/prithvi.yaml \
--dataset_config configs/datasets/biomassters.yaml \
--segmentor_config configs/segmentors/reg_upernet.yaml \
--augmentation_config configs/augmentations/regression_default.yaml \
--num_workers 4 --eval_interval 1 --use_wandb
```

#### Multi-Temporal Regression
```
torchrun ...
torchrun --nnodes=1 --nproc_per_node=1 run.py \
--config configs/run/default.yaml \
--encoder_config configs/foundation_models/prithvi.yaml \
--dataset_config configs/datasets/biomassters.yaml \
--segmentor_config configs/segmentors/reg_upernet_mt.yaml \
--augmentation_config configs/augmentations/regression_default.yaml \
--num_workers 4 --eval_interval 1 --use_wandb
```

### 💻 Fully Supervised Training
Expand Down

0 comments on commit 9fe0f05

Please sign in to comment.