cascade.core.CausalNetwork.training_step

CausalNetwork.training_step(batch, batch_idx)[source]

Training step for a minibatch

Return type:

Tensor