From 378cd72c5f5c9f79fc0248e7b4bc5e1b603adb64 Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Thu, 28 Apr 2022 17:00:15 +0100
Subject: [PATCH] minor change on docs
---
autoaug/autoaugment_learners/AaLearner.py | 11 ++++-------
1 file changed, 4 insertions(+), 7 deletions(-)
diff --git a/autoaug/autoaugment_learners/AaLearner.py b/autoaug/autoaugment_learners/AaLearner.py
index 638b9223..c700bc5e 100644
--- a/autoaug/autoaugment_learners/AaLearner.py
+++ b/autoaug/autoaugment_learners/AaLearner.py
@@ -291,6 +291,7 @@ class AaLearner:
Does the loop which is seen in Figure 1 in the AutoAugment paper
which is:
+
1. <generate a random policy>
2. <see how good that policy is>
3. <save how good the policy is in a list/dictionary and
@@ -310,15 +311,11 @@ class AaLearner:
Args:
train_dataset (torchvision.dataset.vision.VisionDataset):
- test_dataset (torchvision.dataset.vision.VisionDataset):
- child_network_architecture (Union[function, nn.Module]):
- This can be both, for example,
- ``LeNet``
-
- and
+ test_dataset (torchvision.dataset.vision.VisionDataset):
- ``LeNet()``
+ child_network_architecture (Union[function, nn.Module]):
+ This can be both, for example, ``LeNet`` or ``LeNet()``
iterations (int): how many different policies do you want to test
--
GitLab