BaseModule

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

Bases: torch.nn.modules.module.Module

Template for PyTorch modules.

Methods Summary

build_model()	Build model from hparams.
forward(*args, **kwargs)	Push data through module.
freeze()	Prevent updates to module parameters.
unfreeze()	Force updates to module parameters.

Methods Documentation

build_model()

Build model from hparams.

forward(*args, **kwargs)

Push data through module.

freeze()

Prevent updates to module parameters.

unfreeze()

Force updates to module parameters.