diff options
Diffstat (limited to 'recursive.py')
| -rw-r--r-- | recursive.py | 9 |
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: |