Bläddra i källkod

Update run_task.py

Lin Manhui 2 år sedan
förälder
incheckning
afbc0cbd26
1 ändrade filer med 2 tillägg och 2 borttagningar
  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."