Lin Manhui пре 2 година
родитељ
комит
afbc0cbd26
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2 2
      examples/rs_research/run_task.py

+ 2 - 2
examples/rs_research/run_task.py

@@ -70,7 +70,7 @@ if __name__ == '__main__':
             cfg['download_url'], path=cfg['download_path'])
             cfg['download_url'], path=cfg['download_path'])
 
 
     if not isinstance(cfg['datasets']['eval'].args, dict):
     if not isinstance(cfg['datasets']['eval'].args, dict):
-        raise ValueError("args of eval dataset must be a dict!")
+        raise TypeError("args of eval dataset must be a dict!")
     if cfg['datasets']['eval'].args.get('transforms', None) is not None:
     if cfg['datasets']['eval'].args.get('transforms', None) is not None:
         raise ValueError(
         raise ValueError(
             "Found key 'transforms' in args of eval dataset and the value is not None."
             "Found key 'transforms' in args of eval dataset and the value is not None."
@@ -82,7 +82,7 @@ if __name__ == '__main__':
 
 
     if cfg['cmd'] == 'train':
     if cfg['cmd'] == 'train':
         if not isinstance(cfg['datasets']['train'].args, dict):
         if not isinstance(cfg['datasets']['train'].args, dict):
-            raise ValueError("args of train dataset must be a dict!")
+            raise TypeError("args of train dataset must be a dict!")
         if cfg['datasets']['train'].args.get('transforms', None) is not None:
         if cfg['datasets']['train'].args.get('transforms', None) is not None:
             raise ValueError(
             raise ValueError(
                 "Found key 'transforms' in args of train dataset and the value is not None."
                 "Found key 'transforms' in args of train dataset and the value is not None."