gflownet.policy.state_flow

Classes

StateFlow

Takes state in the policy format and predicts its flow (a scalar)

Module Contents

class gflownet.policy.state_flow.StateFlow(config, env, device, float_precision, base=None)[source]

Bases: gflownet.policy.base.ModelBase

Takes state in the policy format and predicts its flow (a scalar)

output_dim = 1[source]
instantiate()[source]
__call__(states)[source]

Returns a tensor of the state flows of the shape (batch_size, )