12345

utils_data.py

@@ -314,7 +314,7 @@ def get_hparams_decode_two_mixture(model_dir=None):
     parser = argparse.ArgumentParser()
     parser.add_argument('-c', '--config', type=str, default="./configs/train_grad.json",
                         help='JSON file for configuration')
-    parser.add_argument('-m', '--model', type=str, required=False, default='/raid/adal_abilbekov/training_emodiff/Emo_diff/logs/logs_train',
+    parser.add_argument('-m', '--model', type=str, required=False, default='.',
                         help='Model name')
     parser.add_argument('-s', '--seed', type=int, default=1234)
     parser.add_argument('--dataset', choices=['train', 'val'], default='val', type=str, help='which dataset to use')