From 5c18be7bda2b42cb470dd406164ac13809530066 Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Mon, 25 Apr 2022 12:59:54 +0100
Subject: [PATCH] make /temp_util to temporarily store stuff while refactoring
is happening
---
.../parse_ds_cn_arch.py | 0
wapp_util.py => temp_util/wapp_util.py | 0
test/MetaAugment/test_ucb_learner.py | 25 +++++++++++++++++++
3 files changed, 25 insertions(+)
rename {MetaAugment/autoaugment_learners => temp_util}/parse_ds_cn_arch.py (100%)
rename wapp_util.py => temp_util/wapp_util.py (100%)
create mode 100644 test/MetaAugment/test_ucb_learner.py
diff --git a/MetaAugment/autoaugment_learners/parse_ds_cn_arch.py b/temp_util/parse_ds_cn_arch.py
similarity index 100%
rename from MetaAugment/autoaugment_learners/parse_ds_cn_arch.py
rename to temp_util/parse_ds_cn_arch.py
diff --git a/wapp_util.py b/temp_util/wapp_util.py
similarity index 100%
rename from wapp_util.py
rename to temp_util/wapp_util.py
diff --git a/test/MetaAugment/test_ucb_learner.py b/test/MetaAugment/test_ucb_learner.py
new file mode 100644
index 00000000..514d7830
--- /dev/null
+++ b/test/MetaAugment/test_ucb_learner.py
@@ -0,0 +1,25 @@
+import MetaAugment.autoaugment_learners as aal
+import MetaAugment.child_networks as cn
+import torch
+import torchvision
+import torchvision.datasets as datasets
+
+import random
+
+
+def test_ucb_learner():
+ policies = UCB1_JC.generate_policies(num_policies, num_sub_policies)
+ q_values, best_q_values = UCB1_JC.run_UCB1(
+ policies,
+ batch_size,
+ learning_rate,
+ ds,
+ toy_size,
+ max_epochs,
+ early_stop_num,
+ iterations,
+ IsLeNet,
+ ds_name
+ )
+ best_q_values = np.array(best_q_values)
+ pass
\ No newline at end of file
--
GitLab