Commit Graph

11 Commits

Author SHA1 Message Date
75475790ac Migrate Dense layer 2024-09-08 13:36:53 +02:00
bc9bff10cd Load running mean and var from weight file 2024-08-25 19:33:33 +02:00
9704d0d53e Add running mean and running var to batchnorm 2024-08-25 19:05:10 +02:00
df47a31f36 Rename dim2d to shape2d 2024-05-27 21:14:51 +02:00
cba177e417 Add getOutputDims to 2d layers 2024-05-26 14:28:43 +02:00
6dca8ccd3c Unify 2d layer naming 2024-05-20 16:23:58 +02:00
74098b24e3 Add support for non square matrices 2024-05-20 15:20:43 +02:00
6f8b5f4081 Rename batchnorm 2024-05-20 13:05:48 +02:00
4a1c4a5f91 Add epsilon param to batch norm 2024-05-19 15:13:22 +02:00
5c8d3f7e25 Compute mean and variance 2024-04-29 20:55:11 +02:00
f60d62f6bd Implement batch norm layer 2024-04-28 19:58:00 +02:00