diff options
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, |