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

4da256a5 · ver217 · GitHub · e67dab92 · 4da256a5
Unverified Commit 4da256a5 authored Dec 02, 2021 by ver217 Committed by GitHub Dec 02, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/vit-b16/vit-b16.py examples/vit-b16/vit-b16.py +1 -1

No files found.
--- 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,
 )