feat: convolution back propapagation + initial model implementation (d…

…enosaurs#21) * feat: neural network api * feat: initial flatten, conv2d, pool2d layer structures * feat: initial dropout layer * chore(rust): rename backend => cpu Co-authored-by: Dean Srebnik <[email protected]> * feat(rust): conv2d init * feat: WIP conv2d layer * conv2d feedforward * feat: max and avg pool api function * fix: tensor types * feat: conv2d wasm example * feat: pool feedforward Co-authored-by: Dean Srebnik <[email protected]> * feat: safetensors dependency * fix: safetensors removed for wasm * fix: safetensors removed * feat(WIP): safe tensors * conv back_propagate Co-authored-by: Dean Srebnik <[email protected]> * feat: save + load model structure set --------- Co-authored-by: CarrotzRule123 <[email protected]> Co-authored-by: Dean Srebnik <[email protected]>
retraigo · May 11, 2023 · b7f43a0 · b7f43a0
1 parent d976616
commit b7f43a0
Show file tree

Hide file tree

Showing 65 changed files with 1,185 additions and 961 deletions.
diff --git a/.gitignore b/.gitignore
@@ -11,4 +11,8 @@ package-lock.json
 test.ts
 experiments
 deno.lock
-target/
+target/
+Cargo.lock
+/target
+
+*.test.bin
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,5 +1,5 @@
 [workspace]
 
 members = [
-    "src/rust"
+    "rust"
 ]