if reinit option is used run runtime rrror to partially iniot the model

2026-09-06 16:30:13 +08:00 · 2019-03-29 17:03:29 +01:00
parent 6edd8bc6dd
commit e1573440ba
1 changed files with 2 additions and 0 deletions
@@ -392,6 +392,8 @@ def main(args):
            # TODO: fix optimizer init, model.cuda() needs to be called before
            # optimizer restore
            # optimizer.load_state_dict(checkpoint['optimizer'])
+            if len(c.reinit_layers) > 0:
+                raise RuntimeError
            model.load_state_dict(checkpoint['model'])
        except:
            print(" > Partial model initialization.")