Spaces:
Sleeping
Sleeping
Updata config.py
Browse files
config.py
CHANGED
|
@@ -30,7 +30,7 @@ class Config:
|
|
| 30 |
self.parser.add_argument('--seed', type=int, default=42, help='随机种子')
|
| 31 |
|
| 32 |
# inference
|
| 33 |
-
self.parser.add_argument('--model_path', type=str, default='
|
| 34 |
self.parser.add_argument('--actions', type=str, default=None, help='指定的魔方动作序列,用空格分隔,如 "U R F D L B"')
|
| 35 |
|
| 36 |
def parse_args(self):
|
|
|
|
| 30 |
self.parser.add_argument('--seed', type=int, default=42, help='随机种子')
|
| 31 |
|
| 32 |
# inference
|
| 33 |
+
self.parser.add_argument('--model_path', type=str, default='/app/checkpoint/final_model_K_30.pth', help='模型路径')
|
| 34 |
self.parser.add_argument('--actions', type=str, default=None, help='指定的魔方动作序列,用空格分隔,如 "U R F D L B"')
|
| 35 |
|
| 36 |
def parse_args(self):
|