oguzakif commited on
Commit
e7aba9f
1 Parent(s): 87cd936

checkpoint paths converted to abs ones

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -75,8 +75,10 @@ parser.add_argument('--Nonlocal', dest='Nonlocal',
75
  default=False, type=bool)
76
 
77
  # RAFT
 
 
78
  parser.add_argument(
79
- '--raft_model', default='../LAFC/flowCheckPoint/raft-things.pth', help="restore checkpoint")
80
  parser.add_argument('--small', action='store_true', help='use small model')
81
  parser.add_argument('--mixed_precision',
82
  action='store_true', help='use mixed precision')
@@ -84,10 +86,13 @@ parser.add_argument('--alternate_corr', action='store_true',
84
  help='use efficent correlation implementation')
85
 
86
  # LAFC
87
- parser.add_argument('--lafc_ckpts', type=str, default='../LAFC/checkpoint')
 
88
 
89
  # FGT
90
- parser.add_argument('--fgt_ckpts', type=str, default='../FGT/checkpoint')
 
 
91
 
92
  # extrapolation
93
  parser.add_argument('--H_scale', dest='H_scale', default=2,
 
75
  default=False, type=bool)
76
 
77
  # RAFT
78
+ # parser.add_argument(
79
+ # '--raft_model', default='../LAFC/flowCheckPoint/raft-things.pth', help="restore checkpoint")
80
  parser.add_argument(
81
+ '--raft_model', default=os.path.abspath(join(project_name, 'FGT_codes', 'LAFC','flowCheckPoint','raft-things.pth')), help="restore checkpoint")
82
  parser.add_argument('--small', action='store_true', help='use small model')
83
  parser.add_argument('--mixed_precision',
84
  action='store_true', help='use mixed precision')
 
86
  help='use efficent correlation implementation')
87
 
88
  # LAFC
89
+ # parser.add_argument('--lafc_ckpts', type=str, default='../LAFC/checkpoint')
90
+ parser.add_argument('--lafc_ckpts', type=str, default=os.path.abspath(join(project_name, 'FGT_codes', 'LAFC','checkpoint')))
91
 
92
  # FGT
93
+ # parser.add_argument('--fgt_ckpts', type=str, default='../FGT/checkpoint')
94
+ parser.add_argument('--fgt_ckpts', type=str, default=os.path.abspath(join(project_name, 'FGT_codes', 'FGT','checkpoint')))
95
+
96
 
97
  # extrapolation
98
  parser.add_argument('--H_scale', dest='H_scale', default=2,