Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
train.sh 498 Bytes
Newer Older
dengjb's avatar
update  
dengjb committed
1
2
3
4
5
6
7
8
9
10 
#!/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