Skip to content

Commit cecea28

Browse files
Vincent Moensfacebook-github-bot
authored andcommitted
[fbsync] Replace --apex with --amp in docs (#5037)
Reviewed By: NicolasHug Differential Revision: D32950944 fbshipit-source-id: b652135dde8cbf0920ee31982fb09471e9c0d2d3
1 parent b21c565 commit cecea28

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

references/video_classification/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ We assume the training and validation AVI videos are stored at `/data/kinectics4
1818

1919
Run the training on a single node with 8 GPUs:
2020
```bash
21-
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
21+
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
2222
```
2323

2424
**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.

0 commit comments

Comments
 (0)