train_single.sh 472 Bytes
Newer Older
Sugon_ldc's avatar
Sugon_ldc committed
1
2
3
4
#!/usr/bin/env bash
export HIP_VISIBLE_DEVICES=0
./tools/dist_train.sh configs/recognition/tsm/tsm_r50_1x1x8_50e_sthv2_rgb.py 1 --validate --seed 0 --cfg-options model.backbone.pretrained=/jiutiandata/7.4/resnet50_8xb32_in1k_20210831-ea4938fc.pth optimizer.lr=0.005 data.videos_per_gpu=8 data.val_dataloader.videos_per_gpu=16 data.test_dataloader.videos_per_gpu=3 data.workers_per_gpu=4 evaluation.interval=1 --log_dir pid.txt 2>&1 | tee tsm_dcu_`date +%Y%m%d%H%M%S`.log