BaseModel

class behavenet.models.base.BaseModel(*args, **kwargs)

Bases: torch.nn.modules.module.Module

Template for PyTorch models.

Methods Summary

build_model()	Build model from hparams.
forward(*args, **kwargs)	Push data through model.
get_parameters()	Get all model parameters that have gradient updates turned on.
loss(*args, **kwargs)	Compute loss.
save(filepath)	Save model parameters.

Methods Documentation

build_model()

Build model from hparams.

forward(*args, **kwargs)

Push data through model.

get_parameters()

Get all model parameters that have gradient updates turned on.

loss(*args, **kwargs)

Compute loss.

save(filepath)

Save model parameters.