OptimizerMigrate

class plugins.train.model._base.io.OptimizerMigrate(config: dict[str, Any], model_path: str)

Bases: object

Migrates weights from a keras optimizer to a torch optimizer’s state dict

Methods Summary

convert()

Convert the keras optimizer from a keras model file into a torch optimizer state dict

Methods Documentation

Parameters:

• config (dict[str, T.Any])

• model_path (str)

convert() → dict[str, Any] | None

Convert the keras optimizer from a keras model file into a torch optimizer state dict

Returns:

• The optimizer state dict for loading into a torch optimizer or None if no saved

• optimizer exists

Return type:

dict[str, Any] | None

convert() → dict[str, Any] | None

Convert the keras optimizer from a keras model file into a torch optimizer state dict

Returns:

• The optimizer state dict for loading into a torch optimizer or None if no saved

• optimizer exists

Return type:

dict[str, Any] | None