-
Notifications
You must be signed in to change notification settings - Fork 13
/
train.py
20 lines (15 loc) · 607 Bytes
/
train.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Copyright Niantic 2019. Patent Pending. All rights reserved.
#
# This software is licensed under the terms of the Monodepth2 licence
# which allows for non-commercial use only, the full terms of which are made
# available in the LICENSE file.
from __future__ import absolute_import, division, print_function
from trainer import Trainer
from options import MonodepthOptions
options = MonodepthOptions()
opts = options.parse()
if __name__ == "__main__":
print("-> model_name: {}".format(opts.model_name))
print("-> split: {}".format(opts.split))
trainer = Trainer(opts)
trainer.train()