Current section

Files

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

include/viva_tensor@distributed@trainer_TrainResult.hrl

-record(train_result, {
final_params :: list(viva_tensor@nn@optim:param()),
final_optimizer :: viva_tensor@nn@optim:optimizer(),
total_loss :: float(),
steps :: integer()
}).