Skip to content

Commit 3068c2d

Browse files
authored
Merge pull request #29 from vinnik-dmitry07/patch-1
2 parents 0d58a76 + f1af6ec commit 3068c2d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/cifar10.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def main():
118118
max_epochs=30,
119119
accelerator="auto",
120120
precision=16,
121+
gradient_clip_val=1.,
121122
devices=1 if torch.cuda.is_available() else None,
122123
logger=CSVLogger(save_dir="logs/"),
123124
callbacks=[SAM()],

0 commit comments

Comments
 (0)