I am trying to load weights from a saved pickle file. It throws an unspecific assertion error.
I was expecting that I can retrieve the opt_state from the saved file. I am not using flax or any other framework.
I am trying to load weights from a saved pickle file. It throws an unspecific assertion error.
I was expecting that I can retrieve the opt_state from the saved file. I am not using flax or any other framework.