From 612956218dcd09987602620f1e2a4147f3d202b7 Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Mon, 25 Apr 2022 12:51:45 +0100
Subject: [PATCH] minor changes
---
MetaAugment/autoaugment_learners/evo_learner.py | 5 ++---
wapp_util.py | 9 ++++++++-
2 files changed, 10 insertions(+), 4 deletions(-)
diff --git a/MetaAugment/autoaugment_learners/evo_learner.py b/MetaAugment/autoaugment_learners/evo_learner.py
index 1ff576b2..18ecf751 100644
--- a/MetaAugment/autoaugment_learners/evo_learner.py
+++ b/MetaAugment/autoaugment_learners/evo_learner.py
@@ -7,12 +7,11 @@ import pygad.torchga as torchga
import copy
import torch
from MetaAugment.controller_networks.evo_controller import evo_controller
-
-from MetaAugment.autoaugment_learners.aa_learner import aa_learner, augmentation_space
import MetaAugment.child_networks as cn
+from .aa_learner import aa_learner, augmentation_space
-class evo_learner():
+class evo_learner(aa_learner):
def __init__(self,
sp_num=1,
diff --git a/wapp_util.py b/wapp_util.py
index 27b51cc4..78be118a 100644
--- a/wapp_util.py
+++ b/wapp_util.py
@@ -12,7 +12,9 @@ import torchvision.datasets as datasets
import MetaAugment.autoaugment_learners as aal
import MetaAugment.controller_networks as cont_n
import MetaAugment.child_networks as cn
-print('@@@ import successful')
+from MetaAugment.main import create_toy
+
+import pickle
def parse_users_learner_spec(
auto_aug_learner,
@@ -30,6 +32,11 @@ def parse_users_learner_spec(
learning_rate,
max_epochs
):
+ """
+ The website receives user inputs on what they want the aa_learner
+ to be. We take those hyperparameters and return an aa_learner
+
+ """
if auto_aug_learner == 'UCB':
policies = aal.ucb_learner.generate_policies(num_policies, num_sub_policies)
q_values, best_q_values = aal.ucb_learner.run_UCB1(
--
GitLab