import fast_depth_functional as fd

if __name__ == '__main__':
    fd.fix_windows_gpu()
    model = fd.mobilenet_nnconv5(weights='imagenet')
    fd.compile(model)
    fd.train(existing_model=model, save_file='../fast-depth-experimental')
    fd.evaluate(model)

    # Save in Tensorflow SavedModel format
    # tf.saved_model.save(model, 'fast_depth_nyu_v2_224_224_3_e1_saved_model')