Explorar o código

Update run_task.py

Lin Manhui %!s(int64=3) %!d(string=hai) anos
pai
achega
afbc0cbd26
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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."