loss curve plot passes on args for log parsing

mu_map/vis/loss_curve.py

@@ -29,8 +29,8 @@ def parse_loss(logs, loss_type):
     return np.array(list(epochs)), np.array(list(losses))
 
 
-def plot_loss(logfile, loss_types, ax, normalize=False, from_epoch=0, to_epoch=None):
-    logs = parse_file(logfile)
+def plot_loss(logfile, loss_types, ax, normalize=False, from_epoch=0, to_epoch=None, start_idx_message=3):
+    logs = parse_file(logfile, start_idx_message=start_idx_message)
     logs = list(filter(lambda logline: logline.loglevel == "INFO", logs))
 
     for i, loss_type in enumerate(loss_types):