EpochChain#

class liesel.goose.chain.EpochChain(*args, **kwargs)[source]#

Bases: Chain[TPyTree]

An EpochChain is a Chain with an associated EpochConfig.

The implementation must implement thinning. That is, if epoch.thinning > 1 and enabled in contructor, the chain must safe only every epoch.thinning element

Methods

append(chunk)

Appends a chunk to the chain.

get()

Returns all chunks combined into one pytree.

Attributes

epoch

Returns the associated EpochConfig.