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:
Piv
2021-03-25 21:28:07 +10:30
parent ab7da5acd4
commit 78d5aace15
2 changed files with 4 additions and 48 deletions

View File

@@ -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')