minor fixes to plotting and logging

locuslab · Nov 29, 2020 · 2bf4384 · 2bf4384
1 parent 76d19e4
commit 2bf4384
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/power_sched/nets.py b/power_sched/nets.py
@@ -12,6 +12,7 @@
 from constants import *
 
 
+
 def task_loss(Y_sched, Y_actual, params):
     return (params["gamma_under"] * torch.clamp(Y_actual - Y_sched, min=0) + 
             params["gamma_over"] * torch.clamp(Y_sched - Y_actual, min=0) + 
@@ -219,11 +220,11 @@ def eval_net(which, model, variables, params, save_folder):
         hold_loss_task = task_loss(
             Y_sched_hold.float(), variables['Y_hold_'], params)
 
-    torch.save(train_loss_task.data, 
+    torch.save(train_loss_task.detach().cpu().numpy(), 
         os.path.join(save_folder, '{}_train_task'.format(which)))
-    torch.save(test_loss_task.data, 
+    torch.save(test_loss_task.detach().cpu().numpy(), 
         os.path.join(save_folder, '{}_test_task'.format(which)))
 
     if (which == "task_net"):
-        torch.save(hold_loss_task.data, 
+        torch.save(hold_loss_task.detach().cpu().numpy(), 
             os.path.join(save_folder, '{}_hold_task'.format(which)))
diff --git a/power_sched/plot.py b/power_sched/plot.py
@@ -55,14 +55,14 @@ def plot_results(load_folders, save_folder):
     colors = [sns.color_palette()[i] for i in [1,2]] + ['gray']
 
     ax = axes[0]
-    ax.set_axis_bgcolor('none')
+    # ax.set_axis_bgcolor('none')
     for col, style, color in zip(rmse_mean.columns, styles, colors):
         rmse_mean[col].plot(
             ax=ax, lw=2, fmt=style, color=color, yerr=rmse_stds[col])
     ax.set_ylabel('RMSE')
 
     ax2 = axes[1]
-    ax2.set_axis_bgcolor('none')
+    # ax2.set_axis_bgcolor('none')
     for col, style, color in zip(task_mean.columns, styles, colors):
         if col == 'Cost-weighted RMSE':
             task_mean[col].plot(