Update utils.py

utils.py

@@ -13,6 +13,7 @@ from torch.utils.data import ConcatDataset
 
 class CharsetMapper(object):
     """A simple class to map ids into strings.
+
     It works only when the character set is 1:1 mapping between individual
     characters and individual ids.
     """
@@ -22,6 +23,7 @@ class CharsetMapper(object):
                  max_length=30,
                  null_char=u'\u2591'):
         """Creates a lookup table.
+
         Args:
           filename: Path to charset file which maps characters to ids.
           max_sequence_length: The max length of ids and string.
@@ -37,8 +39,10 @@ class CharsetMapper(object):
  
     def _read_charset(self, filename):
         """Reads a charset definition from a tab separated text file.
+
         Args:
           filename: a path to the charset file.
+
         Returns:
           a dictionary with keys equal to character codes and values - unicode
           characters.
@@ -196,6 +200,12 @@ class Config(object):
                     self.__setattr__(f'{prefix}{k}', v)
 
         assert os.path.exists(config_path), '%s does not exists!' % config_path
+        with open(config_path) as file:
+            config_dict = yaml.load(file, Loader=yaml.FullLoader)
+        with open('configs/template.yaml') as file:
+            default_config_dict = yaml.load(file, Loader=yaml.FullLoader)
+        __dict2attr(default_config_dict)
+        __dict2attr(config_dict)
         self.global_workdir = os.path.join(self.global_workdir, self.global_name)
 
     def __getattr__(self, item):
@@ -245,6 +255,7 @@ def onehot(label, depth, device=None):
     Args:
         label: shape (n1, n2, ..., )
         depth: a scalar
+
     Returns:
         onehot: (n1, n2, ..., depth)
     """