Skip to content

Commit ee0873a

Browse files
Update vit.py
1 parent d1c5e04 commit ee0873a

File tree

1 file changed

+2
-2
lines changed
  • paddlevideo/modeling/backbones

1 file changed

+2
-2
lines changed

paddlevideo/modeling/backbones/vit.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -380,8 +380,8 @@ def init_weights(self):
380380
if isinstance(
381381
self.pretrained, str
382382
) and self.pretrained.strip() != "": # load pretrained weights
383-
load_ckpt(self, self.pretrained, self.patch_embed.num_patches,
384-
self.seg_num, self.attention_type)
383+
load_ckpt(self, self.pretrained, num_patches=self.patch_embed.num_patches,
384+
seg_num=self.seg_num, attention_type=self.attention_type)
385385
elif self.pretrained is None or self.pretrained.strip() == "":
386386
pass
387387
else:

0 commit comments

Comments
 (0)