Update generate_flow_raft.py

Adding default model argument for RAFT

Author: ponimatkin

generate_flow_raft.py

@@ -76,12 +76,12 @@ def generate_raft_flow(args):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--model', help="restore checkpoint")
+    parser.add_argument('--model', help="restore checkpoint", default="models/raft-sintel.pth")
     parser.add_argument('--dataset', help="dataset for flow estimation")
     parser.add_argument('--step', type=int, default=1, help="flow step size")
     parser.add_argument('--small', action='store_true', help='use small model')
     parser.add_argument('--mixed_precision', action='store_true', help='use mixed precision')
     parser.add_argument('--alternate_corr', action='store_true', help='use efficent correlation implementation')
     args = parser.parse_args()
 
-    generate_raft_flow(args)
+    generate_raft_flow(args)