From 063b20be98747ba0b288545e653ef9489bf14a54 Mon Sep 17 00:00:00 2001 From: y-jan137 Date: Wed, 18 Mar 2026 14:04:47 +0300 Subject: Replace DynMLP with DynNet --- include/train.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'include/train.h') diff --git a/include/train.h b/include/train.h index 9cf613e..94c9666 100644 --- a/include/train.h +++ b/include/train.h @@ -1,6 +1,6 @@ #pragma once -#include "dynmlp.h" +#include "dynnet.h" #include "adam.h" typedef struct { @@ -9,7 +9,7 @@ typedef struct { int nfe_bwd; } TrainStepResult; -TrainStepResult train_step(const DynMLP *net, double *theta, +TrainStepResult train_step(DynNet *net, double *theta, const double **z0s, const double **targets, double t0, double t1, int batch_size, Adam *adam, double atol, double rtol, -- cgit v1.2.3