Loading...
Searching...
No Matches
CNN Class Reference
Implements an architecture consisting of a multi-layer CNN followed by an MLP, according to parameters specified in the input config. More...
Inheritance diagram for CNN:Public Member Functions | |
| __init__ (self, dict config) | |
| torch.Tensor | forward (self, torch.Tensor x) |
| Forward pass through the network. | |
| None | load_cpp_model (self, str model_path) |
| Loads a model saved in cpp jit format. | |
| None | load_model (self, str model_path) |
| Loads a model saved in pytorch format. | |
| Public Member Functions inherited from CNNConfigParser | |
| None | parse (self, dict config) |
| Parse the configuration for the CNN model. | |
Public Attributes | |
| cnn_backbone | |
| mlp | |
| latent_size | |
| linear | |
| Public Attributes inherited from CNNConfigParser | |
| config | |
| input_dim | |
| output_dim | |
| num_layers | |
| latent_size | |
| kernel_size | |
| image_size | |
Detailed Description
Implements an architecture consisting of a multi-layer CNN followed by an MLP, according to parameters specified in the input config.
Constructor & Destructor Documentation
◆ __init__()
| __init__ | ( | self, | |
| dict | config ) |
Reimplemented from CNNConfigParser.
Member Function Documentation
◆ forward()
| torch.Tensor forward | ( | self, | |
| torch.Tensor | x ) |
◆ load_cpp_model()
| None load_cpp_model | ( | self, | |
| str | model_path ) |
◆ load_model()
| None load_model | ( | self, | |
| str | model_path ) |
