[Feature Request] Allow to pass strict=False
when loading model
#116
Labels
enhancement
New feature or request
strict=False
when loading model
#116
In SB2, for
load_parameters
, we had aexact_match
parameter to allow key mismatch when loading.It would be nice to have that, we have to exclude
optimizers
state dict though.This feature allows for instance to load models with same architecture but different names (recently there was a rename with #92 that broke previously saved policy, with
strict=False
we can still load the old models that are still valid).We also had a
custom_objects
parameter to replace parameters on the fly, not sure if it is still needed... @Miffyli could you comment on that?The text was updated successfully, but these errors were encountered: