Hanxiaofeng123 commited on
Commit
fdf1dec
·
verified ·
1 Parent(s): f699756

Updata config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
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='Deepcube/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):
 
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):