crslab.model package¶
Subpackages¶
Submodules¶
-
class
crslab.model.base.
BaseModel
(opt, device, dpath=None, resource=None)[source]¶ Bases:
abc.ABC
,torch.nn.modules.module.Module
Base class for all models
-
converse
(batch, mode)[source]¶ calculate loss and prediction of conversation for batch under certain mode
- Parameters
batch (dict or tuple) – batch data
mode (str, optional) – train/valid/test.
-