elixir-nx / bumblebee

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

Add :spec_overrides to Bumblebee.load_model/2 #340

Closed jonatanklosko closed 4 months ago

jonatanklosko commented 4 months ago
{:ok, spec} = Bumblebee.load_spec({:hf, "microsoft/resnet-50"})
spec = Bumblebee.configure(spec, num_labels: 10)
{:ok, resnet} = Bumblebee.load_model({:hf, "microsoft/resnet-50"}, spec: spec)

->

{:ok, resnet} =
  Bumblebee.load_model({:hf, "microsoft/resnet-50"}, spec_overrides: [num_labels: 10])

I considered accepting a keyword list for :spec, but probably better to kept them separate for clarity.