Remove Experimental model
It didn't perform any better than the regular model Removing batch normalisation significantly harmed training performance
This commit is contained in:
5
main.py
5
main.py
@@ -2,7 +2,10 @@ import fast_depth_functional as fd
|
||||
|
||||
if __name__ == '__main__':
|
||||
fd.fix_windows_gpu()
|
||||
model = fd.mobilenet_nnconv5_no_bn(weights='imagenet')
|
||||
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')
|
||||
|
||||
Reference in New Issue
Block a user