Skip to content

Commit 6a9d245

Browse files
author
Mathilde Caron
committed
typo clip grad without fp16
1 parent d7eea5a commit 6a9d245

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main_dino.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def train_one_epoch(student, teacher, teacher_without_ddp, dino_loss, data_loade
313313
if fp16_scaler is None:
314314
loss.backward()
315315
if args.clip_grad:
316-
param_norms = utils.clip_gradients(model, args.clip_grad)
316+
param_norms = utils.clip_gradients(student, args.clip_grad)
317317
utils.cancel_gradients_last_layer(epoch, student,
318318
args.freeze_last_layer)
319319
optimizer.step()

0 commit comments

Comments
 (0)