Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
TSM_pytorch
Commits
d8e27f85
Commit
d8e27f85
authored
Nov 22, 2023
by
Sugon_ldc
Browse files
add pertrain model adress
parent
26f78371
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
2 deletions
+4
-2
README.md
README.md
+2
-0
train.sh
train.sh
+1
-1
train_single.sh
train_single.sh
+1
-1
No files found.
README.md
View file @
d8e27f85
...
...
@@ -50,6 +50,8 @@ mmcv-full:1.6.1+gitdebbc80.dtk2210
https://pan.baidu.com/s/1NCqL7JVoFZO6D131zGls-A#list/path=%2F&parentPath=%2Fsharelink3725532333-553839935448680 密码:07ka
预训练模型文件:
[
resnet50_8xb32_in1k_20210831-ea4938fc.pth
](
https://download.openmmlab.com/mmclassification/v0/resnet/resnet50_8xb32_in1k_20210831-ea4938fc.pth
)
### 数据预处理
进入工程目录后,创建data子目录,进入data目录,创建软链接指向数据集实际路径:
...
...
train.sh
View file @
d8e27f85
#!/usr/bin/env bash
export
HIP_VISIBLE_DEVICES
=
0,1,2,3
./tools/dist_train.sh configs/recognition/tsm/tsm_r50_1x1x8_50e_sthv2_rgb.py 4
--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
./tools/dist_train.sh configs/recognition/tsm/tsm_r50_1x1x8_50e_sthv2_rgb.py 4
--validate
--seed
0
--cfg-options
model.backbone.pretrained
=
.
/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
train_single.sh
View file @
d8e27f85
#!/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
./tools/dist_train.sh configs/recognition/tsm/tsm_r50_1x1x8_50e_sthv2_rgb.py 1
--validate
--seed
0
--cfg-options
model.backbone.pretrained
=
.
/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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment