Fix path issue in second plot of stage1

code/compute_results.py

@@ -188,7 +188,7 @@ if __name__ == "__main__":
             title='Loss values of {} using the best hyperparams'.format(args.dataset_name)
         )
         Plotter.plot_losses(
-            file_path=output_path + 'stage1' + os.sep + 'losses_wo_params.png',
+            file_path=output_path + os.sep + 'losses_wo_params.png',
             all_experiment_scores=[base_wo_params_train_scores, base_wo_params_dev_scores, base_wo_params_test_scores,
                 random_wo_params_train_scores, random_wo_params_dev_scores, random_wo_params_test_scores,
                 omp_wo_params_train_scores, omp_wo_params_dev_scores, omp_wo_params_test_scores],