Skip to content

SheepHuan/3DHPE-Benchmark

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

3DHPE-Benchmark

CUDA_VISIBLE_DEVICES=0,1 torchrun --nproc_per_node=2 pylib/train.py
CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --nproc_per_node=4 pylib/train.py
onnxsim --input-shape="input:1,3,640,640" tmp/mature/rtmo_s.onnx tmp/mature/rtmo_s.onnx
mnnconvert --framework ONNX --modelFile tmp/mature/rtmo_l_640x640.onnx --fp16 --MNNModel tmp/mature/rtmo_l_640x640.mnn
mnnconvert --framework ONNX --modelFile tmp/mature/rtmo_s.onnx --fp16 --saveStaticModel --MNNModel tmp/mature/rtmo_s.mnn

mnnconvert --framework ONNX --modelFile tmp/mature/dw-ll_ucoco_384x288.onnx --fp16 --MNNModel tmp/mature/dw-ll_ucoco_384x288.mnn

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages