Update tests for torch/cuda

2021-03-30 12:05:52 +00:00
parent c2270fd153
commit e5ec43e04a
12 changed files with 386 additions and 248 deletions
--- a/lib/config_utils/init.py
+++ b/lib/config_utils/init.py
@@ -3,6 +3,7 @@
 ##################################################
 # general config related functions
 from .config_utils import load_config, dict2config, configure2str
+
 # the args setting for different experiments
 from .basic_args import obtain_basic_args
 from .attention_args import obtain_attention_args