Snow-White-995 commited on
Commit
3455f74
·
1 Parent(s): d4bb6e8

Update infer.py

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -119,7 +119,7 @@ def main():
119
  parser = argparse.ArgumentParser()
120
  parser.add_argument('--config', type=str, default='configs/config.json')
121
  parser.add_argument('--device', type=str, default='cpu')
122
- parser.add_argument('--model_path', type=str, default='models/G_1000.pth')
123
  parser.add_argument('--output', type=str, default='sample')
124
  args = parser.parse_args()
125
 
 
119
  parser = argparse.ArgumentParser()
120
  parser.add_argument('--config', type=str, default='configs/config.json')
121
  parser.add_argument('--device', type=str, default='cpu')
122
+ parser.add_argument('--model_path', type=str, default='models/G_73000.pth')
123
  parser.add_argument('--output', type=str, default='sample')
124
  args = parser.parse_args()
125