From cbee22bf6cfca8ecc45ceef62944b81e2c3e3903 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sun, 17 Jun 2018 09:26:48 +0200
Subject: args

---
 recursive.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'recursive.py')

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:
-- 
cgit v1.2.3-70-g09d2