master

分支 (33)

标签 (27)

管理

管理

master

bits_and_tpu

more_datasets

fixes_bce_regnet

attn_update

opt_cleanup

perceiver

vit_and_bit_test_fixes

cleanup_xla_model_fixes

more_attn

levit_visformer_rednet

more_mlp

effnetv2_official

cait

mlp-mixer

byoanet-self_attn

pit_and_vit_update

benchmark-fixes-vit_hybrids

hf_hub_revisit

gh-pages

v0.1-rsb-weights

v0.1-attn-weights

v0.4.12

v0.1-vt3p-weights

v0.4.9

v0.1-effv2-weights

v0.1-rs-weights

v0.1-coat-weights

v0.1-pit-weights

v0.4.5

v0.1-dnf-weights

v0.1-ger-weights

v0.1-repvgg-weights

v0.3.4

v0.3.3

v0.1-vitjx

v0.2.1

v0.1-rexnet

v0.1-resnest

v0.1-regnet

pytorch-image-models
/
distributed_train.sh

#!/bin/bash
NUM_PROC=$1
shift
python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@"