From 00cd5b38f26e40fbb0dd21e9682e9a9580fe3e78 Mon Sep 17 00:00:00 2001 From: y-jan137 Date: Wed, 18 Mar 2026 09:56:29 +0300 Subject: Reformat --- include/train.h | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 include/train.h (limited to 'include/train.h') diff --git a/include/train.h b/include/train.h new file mode 100644 index 0000000..9cf613e --- /dev/null +++ b/include/train.h @@ -0,0 +1,16 @@ +#pragma once + +#include "dynmlp.h" +#include "adam.h" + +typedef struct { + double loss; + int nfe_fwd; + int nfe_bwd; +} TrainStepResult; + +TrainStepResult train_step(const DynMLP *net, double *theta, + const double **z0s, const double **targets, + double t0, double t1, int batch_size, + Adam *adam, double atol, double rtol, + int num_checkpoints); -- cgit v1.2.3