fix get_loss_fun() err message (#181)

Summary: Pull Request resolved: #181 Reviewed By: sdebnathusc Differential Revision: D36088827 Pulled By: vaibhava0 fbshipit-source-id: 7c3f14bcb42329af7dd188bb9978627c8724a121
facebookresearch · Jun 3, 2022 · ce77f9e · ce77f9e
1 parent 5a7442e
commit ce77f9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycls/core/builders.py b/pycls/core/builders.py
@@ -40,7 +40,7 @@ def get_model():
 def get_loss_fun():
     """Gets the loss function class specified in the config."""
     err_str = "Loss function type '{}' not supported"
-    assert cfg.MODEL.LOSS_FUN in _loss_funs.keys(), err_str.format(cfg.TRAIN.LOSS)
+    assert cfg.MODEL.LOSS_FUN in _loss_funs.keys(), err_str.format(cfg.MODEL.LOSS_FUN)
     return _loss_funs[cfg.MODEL.LOSS_FUN]