Snow-White-995
commited on
Commit
·
3455f74
1
Parent(s):
d4bb6e8
Update infer.py
Browse files
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/
|
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 |
|