Hugging Face's logo

Spaces:
JohnTitor22
/
Unimernet_gradio
Running

App Files Community
Unimernet_gradio
File size: 507 Bytes 
12b7ec5
 
 
 
 
 
 
 
 
 
 
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