From b529d130df7b175579075ab1574a66fa77e9b319 Mon Sep 17 00:00:00 2001 From: Si11ium Date: Thu, 21 May 2020 12:33:36 +0200 Subject: [PATCH] module_mixins.py updated with tensor return --- multi_run.py | 2 +- util/module_mixins.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/multi_run.py b/multi_run.py index 1eb79be..85cede6 100644 --- a/multi_run.py +++ b/multi_run.py @@ -46,7 +46,7 @@ if __name__ == '__main__': data_noise_ratio=0.0, data_shift_ratio=0.0, data_loudness_ratio=0.4, data_stretch=True, train_epochs=101) - for dicts in [raw_conf, all_conf, speed_conf, mask_conf,noise_conf, shift_conf, loudness_conf]: + for dicts in [raw_conf, all_conf, speed_conf, mask_conf, noise_conf, shift_conf, loudness_conf]: arg_dict.update(dicts) config = config.update(arg_dict) diff --git a/util/module_mixins.py b/util/module_mixins.py index 13fff52..0f786d0 100644 --- a/util/module_mixins.py +++ b/util/module_mixins.py @@ -100,7 +100,7 @@ class BaseValMixin: uar_score = sklearn.metrics.recall_score(y_true, y_pred, labels=[0, 1], average='macro', sample_weight=None, zero_division='warn') - + uar_score = torch.as_tensor(uar_score) summary_dict['log'].update({f'uar{ident}_score': uar_score}) return summary_dict