Skip to content

Commit

Permalink
Update resnet_encoder.py
Browse files Browse the repository at this point in the history
Fix as suggested in comment Uehwan#4 (comment)
  • Loading branch information
sreejithpai committed Mar 7, 2022
1 parent 288ae6f commit 45641ab
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion networks/resnet_encoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def __init__(
cfg, self.maskrcnn, save_dir='.'
)
if pretrained_model_path is not None:
_ = self.checkpointer.load(pretrained_model_path, convert=True)
_ = self.checkpointer.load(pretrained_model_path, convert=False)

if not self.joint_training:
# freeze gradients for mask rcnn
Expand Down

0 comments on commit 45641ab

Please sign in to comment.