scripts/train.sh

#!/bin/bash

# train unimernet_base
python -m torch.distributed.launch --nproc_per_node=8 train.py --cfg-path configs/train/unimernet_base_encoder6666_decoder8_dim1024.yaml

# train unimernet_small
# python -m torch.distributed.launch --nproc_per_node=8 train.py --cfg-path configs/train/unimernet_small_encoder6666_decoder8_dim768.yaml

# train unimernet_tiny
# python -m torch.distributed.launch --nproc_per_node=8 train.py --cfg-path configs/train/unimernet_tiny_encoder6666_decoder8_dim768.yaml