From c2395ee13fbd1515eea87c42cc96e3cab27bdc01 Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Sat, 19 Feb 2022 17:43:38 +0000
Subject: [PATCH] Add demo code for
MetaAugment.autoaugment_learners.autoaugment
---
MetaAugment/autoaugment_learners/autoaugment.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/MetaAugment/autoaugment_learners/autoaugment.py b/MetaAugment/autoaugment_learners/autoaugment.py
index e5ee9f5e..977946bc 100644
--- a/MetaAugment/autoaugment_learners/autoaugment.py
+++ b/MetaAugment/autoaugment_learners/autoaugment.py
@@ -409,6 +409,7 @@ if __name__=='__main__':
from MetaAugment.main import *
import MetaAugment.child_networks as cn
import torchvision.transforms as transforms
+ from torchvision.transforms import functional as F, InterpolationMode
batch_size = 32
n_samples = 0.005
@@ -435,7 +436,7 @@ if __name__=='__main__':
aa_transform.policies = policies
train_transform = transforms.Compose([
- aa_transform(),
+ aa_transform,
transforms.ToTensor()
])
@@ -455,4 +456,4 @@ if __name__=='__main__':
test_autoaugment_policy(policies1)
- test_autoaugment_policy(policies2)
+ test_autoaugment_policy(policies2)
\ No newline at end of file
--
GitLab