From b065e36f1b8c39ea597743f74aa79d09b241d424 Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Mon, 13 May 2019 20:02:48 +0200 Subject: [PATCH] Fix double name exception --- ulmfit/__main__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ulmfit/__main__.py b/ulmfit/__main__.py index 546ba88..93e8b8a 100644 --- a/ulmfit/__main__.py +++ b/ulmfit/__main__.py @@ -123,8 +123,9 @@ class ULMFiT: **kwargs) - def poleval19_seeds(self, base, seed_name='clsweightseed', model_num=10, **kwargs): - name = folder_name_to_model_name(Path(base).name) + def poleval19_seeds(self, base, name=None, seed_name='clsweightseed', model_num=10, **kwargs): + if name is None: + name = folder_name_to_model_name(Path(base).name) for seed in range(0, model_num, 1): kwargs[seed_name] = seed print("Seed: ", seed_name, seed)