add some details in vit-b16 example (#46)

2025-08-31 16:40:41 +00:00 · 2021-12-02 09:29:27 +08:00
parent e67dab92a9
commit 4da256a584
1 changed files with 1 additions and 1 deletions
--- a/examples/vit-b16/vit-b16.py
+++ b/examples/vit-b16/vit-b16.py
@@ -73,6 +73,6 @@ dali = dict(
 engine = dict(
    schedule=None,
    gradient_handlers=None,
-    gradient_accumulation=16,
+    gradient_accumulation=32,
    gradient_clipping=1.0,
 )