Євген

icon indicating a website link https://3bytesahead.com/ icon indicating an email [email protected]

icon company Technical University of Denmark icon location Copenhagen, Denmark icon location 🇺🇦 Привіт! I'm a postdoc at DTU Compute. Founder of 3bytesahead - a Copenhagen based full stack data science consultancy.

Results 3 comments of Євген

Pickling a trained model (NNX)

Hey @cgarciae , thanks for swift reply! It might work, but still an extra dependency. It feels like "wholesome" save/load functionality is so fundamental, it should be added to NNX...

Pickling a trained model (NNX)

@cgarciae I get your point. But even with `orbax`, the attributes of the model are still stored saparately. Let's take a step back. The very premise of Flax NNX (and...

Pickling a trained model (NNX)

@cgarciae thats a step in the right direction, but I would love to hear your deeper considerations. To require `cloudpickle`, a separate dependency just to save, what suppose to be...