Load model weights from checkpoint in non-strict mode

Summary: Updated load_from_checkpoint method call inside lighting_task.py to include extra 'strict' keyword parameter

Reviewed By: kazhang

Differential Revision: D29446372

fbshipit-source-id: b14bc13db551f0876ca78d3ea164cfb08e71a757

d2go/runner/lightning_task.py

@@ -153,7 +153,7 @@ class DefaultTask(pl.LightningModule):
         if cfg.MODEL.WEIGHTS:
             # only load model weights from checkpoint
             logger.info(f"Load model weights from checkpoint: {cfg.MODEL.WEIGHTS}.")
-            task = cls.load_from_checkpoint(cfg.MODEL.WEIGHTS, cfg=cfg)
+            task = cls.load_from_checkpoint(cfg.MODEL.WEIGHTS, cfg=cfg, strict=False)
         else:
             task = cls(cfg)