Current section

Files

Jump to
viva_tensor include viva_tensor@distributed@trainer_TrainConfig.hrl
Raw

include/viva_tensor@distributed@trainer_TrainConfig.hrl

-record(train_config, {
num_workers :: integer(),
batches_per_step :: integer(),
grad_aggregation :: viva_tensor@distributed@trainer:grad_aggregation()
}).