78d5aace15
It didn't perform any better than the regular model Removing batch normalisation significantly harmed training performance
12 lines
384 B
Python
12 lines
384 B
Python
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')
|