afe.ir.serializer
Submodules
Functions
|
Loads an AwesomeNet from yaml and npz files |
|
Saves an AwesomeNet to a yaml and npz file. Overwrites existing files. |
Package Contents
- afe.ir.serializer.load_awesomenet(model_name: str, network_directory: str = '') afe.ir.net.AwesomeNet [source]
Loads an AwesomeNet from yaml and npz files
- Parameters:
model_name – The name of the model, used in file names.
network_directory – The directory to load the files from.
- afe.ir.serializer.save_awesomenet(net: afe.ir.net.AwesomeNet, model_name: str, output_directory: str = '') None [source]
Saves an AwesomeNet to a yaml and npz file. Overwrites existing files.
- Parameters:
net – AwesomeNet to save.
model_name – The name of the model, used in file names.
output_directory – The directory to save the files to.