Weights

class plugins.train.model._base.io.Weights(plugin: ModelBase)

Bases: object

Handling of freezing and loading model weights

Parameters:: plugin (ModelBase) – The parent plugin class that owns the IO functions.

Methods Summary

`freeze`()	If freeze has been selected in the cli arguments, then freeze those models indicated in the plugin's configuration.
`load`(model_exists)	Load weights for newly created models, or output warning for pre-existing models.

Methods Documentation

freeze() → None

If freeze has been selected in the cli arguments, then freeze those models indicated in the plugin’s configuration.

load(model_exists: bool) → None

Load weights for newly created models, or output warning for pre-existing models.

Parameters:: model_exists (bool) – True if a model pre-exists and is being resumed, False if this is a new model
Return type:: None

freeze() → None

If freeze has been selected in the cli arguments, then freeze those models indicated in the plugin’s configuration.

load(model_exists: bool) → None

Load weights for newly created models, or output warning for pre-existing models.

Parameters:: model_exists (bool) – True if a model pre-exists and is being resumed, False if this is a new model
Return type:: None