Unable to save keras model with multi worker distribution strategy

TensorFlow General Discussion
, ,
1

https://stackoverflow.com/questions/72082204/unable-to-save-keras-model-with-multi-worker-distribution-strategy

3

Hi @rishabhC, For saving the model using the distributed strategy the saving destination needs to be different for each worker.

  • For non-chief workers, you will need to save the model to a temporary directory.
  • For the chief, you will need to save to the provided model directory.

please refer to this document for more details. Thank You.