Model
srai.embedders.Model ¶
Bases: LightningModule
Class for model based on LightningModule.
get_config ¶
Get model config.
Source code in srai/embedders/_base.py
load ¶
classmethod
Load model from a file.
PARAMETER | DESCRIPTION |
---|---|
path
|
Path to the file.
TYPE:
|
**kwargs
|
Additional kwargs to pass to the model constructor.
TYPE:
|
Source code in srai/embedders/_base.py
save ¶
Save the model to a directory.
PARAMETER | DESCRIPTION |
---|---|
path
|
Path to the directory.
TYPE:
|