Loading...
Searching...
No Matches
LPAC Class Reference
LPAC neural network architecture. More...
Inheritance diagram for LPAC:Public Member Functions | |
| __init__ (self, in_config) | |
| torch.Tensor | forward (self, torch_geometric.data.Data data) |
| Forward pass of the LPAC model. | |
| None | load_model (self, str model_state_dict_path) |
| Load the model from the state dict. | |
| None | load_cnn_backbone (self, str model_path) |
| Load the CNN backbone from the model path. | |
| None | load_gnn_backbone (self, str model_path) |
| Load the GNN backbone from the model path. | |
| Public Member Functions inherited from GNNConfigParser | |
| None | parse (self, dict config) |
| Parse the configuration for the GNN model. | |
Public Attributes | |
| cnn_config | |
| cnn_backbone | |
| gnn_backbone | |
| gnn_mlp | |
| output_linear | |
| output_dim | |
| Public Attributes inherited from GNNConfigParser | |
| config | |
| input_dim | |
| output_dim | |
| num_hops | |
| num_layers | |
| latent_size | |
Detailed Description
Constructor & Destructor Documentation
◆ __init__()
| __init__ | ( | self, | |
| in_config ) |
Reimplemented from GNNConfigParser.
Member Function Documentation
◆ forward()
| torch.Tensor forward | ( | self, | |
| torch_geometric.data.Data | data ) |
◆ load_cnn_backbone()
| None load_cnn_backbone | ( | self, | |
| str | model_path ) |
◆ load_gnn_backbone()
| None load_gnn_backbone | ( | self, | |
| str | model_path ) |
◆ load_model()
| None load_model | ( | self, | |
| str | model_state_dict_path ) |
