From 30bb1be841b7e7d0c60bfba8177795f32bdd5873 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 7 Dec 2021 00:21:20 +0000 Subject: [PATCH] Replace --apex with --amp in docs --- references/video_classification/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/references/video_classification/README.md b/references/video_classification/README.md index 644d4b2409d..204bfda5ba5 100644 --- a/references/video_classification/README.md +++ b/references/video_classification/README.md @@ -18,7 +18,7 @@ We assume the training and validation AVI videos are stored at `/data/kinectics4 Run the training on a single node with 8 GPUs: ```bash -torchrun --nproc_per_node=8 train.py --data-path=/data/kinectics400 --train-dir=train --val-dir=val --batch-size=16 --cache-dataset --sync-bn --apex +torchrun --nproc_per_node=8 train.py --data-path=/data/kinectics400 --train-dir=train --val-dir=val --batch-size=16 --cache-dataset --sync-bn --amp ``` **Note:** all our models were trained on 8 nodes with 8 V100 GPUs each for a total of 64 GPUs. Expected training time for 64 GPUs is 24 hours, depending on the storage solution.