Hi, I’m new to Tensorflow and even Python. I’m interested in to use Tensorflow3D models to recognize something on mobile devices.
Before starting my problem question, can I use a tensorflow3D model on a mobile device in the first place? If so, I have a problem.
My problem is that I can’t save a model that is using semantic segmentation model, named “semantic_segmentation_model.” I’m struggled with saving the model just adding below code.
model.fit( x=inputs, callbacks=[backup_checkpoint_callback, checkpoint_callback], steps_per_epoch=FLAGS.num_steps_per_epoch, epochs=FLAGS.num_epochs, verbose=1 if FLAGS.run_functions_eagerly else 2) model.close_writer() # Added this by edom18 model.save("fileName")
This added code is in
google-reserch/tf3d/train.py. I just added last code after calling
fit method. Then I got some errors like below.
ValueError: batch_size is unknown at graph construction time. In call to configurable 'train' (<function train at 0x7f3e476aee60>)
What should I know about that?