You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Which currently fails. The reason is that there are a range of checks which use get_model_name(), a function that accesses the class name. So that function needs to be changed and different PyTorch networks renamed in the MODEL_REGISTRY. This means that a few methods/functions will likely need an additional input which is a dict mapping model names. See also #175
The text was updated successfully, but these errors were encountered:
In the MODEL_REGISTRY under
autoemulate.emulators.__init__py
, we are adding models like this:A second neural network would be added like this:
Which currently fails. The reason is that there are a range of checks which use
get_model_name()
, a function that accesses the class name. So that function needs to be changed and different PyTorch networks renamed in theMODEL_REGISTRY
. This means that a few methods/functions will likely need an additional input which is a dict mapping model names. See also #175The text was updated successfully, but these errors were encountered: