rcognita.models.ModelWeightContainer

class rcognita.models.ModelWeightContainer(dim_output, weights_init=None)

Trivial model, which is typically used in actor in which actions are being optimized directly.

__init__(dim_output, weights_init=None)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(dim_output[, weights_init])	Initialize self.
cache_weights([weights])
forward(*argin[, weights])
restore_weights()	Assign the weights of the cached model to the active model.
update_and_cache_weights(weights)
update_weights(weights)

Attributes

model_name