9 lines
267 B
Python
9 lines
267 B
Python
import fast_depth_functional as fd
|
|
|
|
if __name__ == '__main__':
|
|
fd.fix_windows_gpu()
|
|
model = fd.mobilenet_nnconv5_no_bn(weights='imagenet')
|
|
fd.compile(model)
|
|
fd.train(existing_model=model, save_file='../fast-depth-experimental')
|
|
fd.evaluate(model)
|