args

1 files changed, 9 insertions, 0 deletions

diff --git a/recursive.py b/recursive.py
index 61c5f62..461f3e1 100644
--- a/recursive.py
+++ b/recursive.py
@@ -43,6 +43,15 @@ dataset = data_loader.load_data()
 start_img_path = os.path.join(render_dir, "frame_00000.png")
 copyfile(opt.start_img, start_img_path)
 
+if not opt.engine and not opt.onnx:
+    model = create_model(opt)
+    if opt.data_type == 16:
+        model.half()
+    elif opt.data_type == 8:
+        model.type(torch.uint8)
+    if opt.verbose:
+        print(model)
+
 for i, data in enumerate(dataset):
     print("iter...{}".format(i))
     if i >= opt.how_many: