jhtonyKoo commited on
Commit
1f4e672
1 Parent(s): e535c7a

Update inference/style_transfer.py

Browse files
Files changed (1) hide show
  1. inference/style_transfer.py +1 -1
inference/style_transfer.py CHANGED
@@ -329,7 +329,7 @@ def set_up():
329
 
330
  directory_args = parser.add_argument_group('Directory args')
331
  # directory paths
332
- directory_args.add_argument('--target_dir', type=str, default='./yt_dir/')
333
  directory_args.add_argument('--output_dir', type=str, default=None, help='if no output_dir is specified (None), the results will be saved inside the target_dir')
334
  directory_args.add_argument('--input_file_name', type=str, default='input')
335
  directory_args.add_argument('--reference_file_name', type=str, default='reference')
 
329
 
330
  directory_args = parser.add_argument_group('Directory args')
331
  # directory paths
332
+ directory_args.add_argument('--target_dir', type=str, default='./')
333
  directory_args.add_argument('--output_dir', type=str, default=None, help='if no output_dir is specified (None), the results will be saved inside the target_dir')
334
  directory_args.add_argument('--input_file_name', type=str, default='input')
335
  directory_args.add_argument('--reference_file_name', type=str, default='reference')