From 3f0557f2fcd24845f5b5530e5837398c64cdfbc0 Mon Sep 17 00:00:00 2001
From: John Carter <jac202@ic.ac.uk>
Date: Thu, 21 Apr 2022 09:39:41 +0100
Subject: [PATCH] UCB1 set for evaluation
---
MetaAugment/UCB1_JC.ipynb | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/MetaAugment/UCB1_JC.ipynb b/MetaAugment/UCB1_JC.ipynb
index 2ebf29b8..1c573974 100644
--- a/MetaAugment/UCB1_JC.ipynb
+++ b/MetaAugment/UCB1_JC.ipynb
@@ -441,7 +441,7 @@
"\n",
"batch_size = 32 # size of batch the inner NN is trained with\n",
"learning_rate = 1e-1 # fix learning rate\n",
- "ds = \"MNIST\" # pick dataset (MNIST, KMNIST, FashionMNIST, CIFAR10, CIFAR100)\n",
+ "ds = \"FashionMNIST\" # pick dataset (MNIST, KMNIST, FashionMNIST, CIFAR10, CIFAR100)\n",
"toy_size = 1 # total propeortion of training and test set we use\n",
"max_epochs = 100 # max number of epochs that is run if early stopping is not hit\n",
"early_stop_num = 10 # max number of worse validation scores before early stopping is triggered\n",
--
GitLab