wassname/multifit
1
0
Fork 0
mirror of https://github.com/wassname/multifit.git synced 2026-09-09 11:27:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix double name exception

Browse Source
This commit is contained in:
Piotr Czapla
2019-05-13 20:02:48 +02:00
parent 9101a0313c
commit b065e36f1b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/__main__.py
+3 -2
View File
@@ -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)