From dc3ea6827478f753dd77751acae253fb696c0a06 Mon Sep 17 00:00:00 2001
From: Max Ramsay King <maxramsayking@gmail.com>
Date: Mon, 25 Apr 2022 16:17:15 +0100
Subject: [PATCH] updated aa_learner
---
MetaAugment/autoaugment_learners/aa_learner.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/MetaAugment/autoaugment_learners/aa_learner.py b/MetaAugment/autoaugment_learners/aa_learner.py
index 75f01f8c..29b5d29a 100644
--- a/MetaAugment/autoaugment_learners/aa_learner.py
+++ b/MetaAugment/autoaugment_learners/aa_learner.py
@@ -101,7 +101,7 @@ class aa_learner:
self.history = []
self.augmentation_space = [x for x in augmentation_space if x not in exclude_method]
self.fun_num = len(augmentation_space)
- self.op_tensor_length = self.fun_num +p_bins+m_bins if discrete_p_m else self.fun_num +2
+ self.op_tensor_length = self.fun_num + p_bins + m_bins if discrete_p_m else self.fun_num +2
def translate_operation_tensor(self, operation_tensor, return_log_prob=False, argmax=False):
--
GitLab