We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d1c5e04 commit ee0873aCopy full SHA for ee0873a
paddlevideo/modeling/backbones/vit.py
@@ -380,8 +380,8 @@ def init_weights(self):
380
if isinstance(
381
self.pretrained, str
382
) 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)
+ load_ckpt(self, self.pretrained, num_patches=self.patch_embed.num_patches,
+ seg_num=self.seg_num, attention_type=self.attention_type)
385
elif self.pretrained is None or self.pretrained.strip() == "":
386
pass
387
else:
0 commit comments