Closed nathanin closed 3 years ago
https://github.com/tensorflow/tensorflow/issues/37862
This has to be done in the notebook/script using the main MoCo class:
MoCo
https://github.com/nathanin/micron2/blob/1cd935450581fba555ca4e27cceeadac0cef1d46/micron2/clustering/model_moco.py#L91-L93
https://github.com/tensorflow/tensorflow/issues/37862