diff options
| author | y-jan137 <yousefjan24000@gmail.com> | 2026-03-18 14:04:47 +0300 |
|---|---|---|
| committer | y-jan137 <yousefjan24000@gmail.com> | 2026-03-18 14:04:47 +0300 |
| commit | 063b20be98747ba0b288545e653ef9489bf14a54 (patch) | |
| tree | b655758d2361dda0c6ffd35d6add6c5ea0f5aba4 /include/train.h | |
| parent | c9583c9897120cea7357ebc78b621aac0440da2c (diff) | |
Replace DynMLP with DynNet
Diffstat (limited to 'include/train.h')
| -rw-r--r-- | include/train.h | 4 |
1 files changed, 2 insertions, 2 deletions
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, |