search

elixir-nx / bumblebee

Pre-trained Neural Network models in Axon (+ 🤗 Models integration)
Apache License 2.0
1.26k stars 90 forks source link

Use model state #375

Open seanmor5 opened 1 month ago

seanmor5 commented 1 month ago

Resolves #372

jonatanklosko commented 1 month ago

@seanmor5 model_state = Axon.trace_init(model, input_template) this still warns because it expects the struct, should there be a function to build the struct from the map perhaps?

(we capture logs in most tests, so it doesn't show up there)