From 23de35b2546e6c2c20ce7756c00a84f28f4270a5 Mon Sep 17 00:00:00 2001 From: Erik Pettersson Date: Thu, 16 May 2024 14:11:10 +0200 Subject: [PATCH] Update with correct flags --- configs/skeleton/posec3d/custom_dataset_training.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/skeleton/posec3d/custom_dataset_training.md b/configs/skeleton/posec3d/custom_dataset_training.md index 81fc1cb3e1..eefef9d9dd 100644 --- a/configs/skeleton/posec3d/custom_dataset_training.md +++ b/configs/skeleton/posec3d/custom_dataset_training.md @@ -15,7 +15,7 @@ We provide a step-by-step tutorial on how to train your custom dataset with Pose 2. Then, you will collect all the pickle files into one list for training (and, of course, for validation) and save them as a single file (like `custom_dataset_train.pkl` or `custom_dataset_val.pkl`). At that time, you finalize preparing annotation files for your custom dataset. -3. Next, you may use the following script (with some alterations according to your needs) for training as shown in [PoseC3D/Train](https://github.com/open-mmlab/mmaction2/blob/master/configs/skeleton/posec3d/README.md#train): `python tools/train.py configs/skeleton/posec3d/slowonly_r50_u48_240e_ntu120_xsub_keypoint.py --work-dir work_dirs/slowonly_r50_u48_240e_ntu120_xsub_keypoint --validate --test-best --gpus 2 --seed 0 --deterministic`: +3. Next, you may use the following script (with some alterations according to your needs) for training as shown in [PoseC3D/Train](https://github.com/open-mmlab/mmaction2/blob/master/configs/skeleton/posec3d/README.md#train): `python tools/train.py configs/skeleton/posec3d/slowonly_r50_u48_240e_ntu120_xsub_keypoint.py --work-dir work_dirs/slowonly_r50_u48_240e_ntu120_xsub_keypoint --seed 0 --deterministic`: - Before running the above script, you need to modify the variables to initialize with your newly made annotation files: