diff --git a/README.md b/README.md index 5620253..e8d6d96 100644 --- a/README.md +++ b/README.md @@ -1,13 +1,20 @@ # burn-examples some neural networks implemented by burn framework: -## 1. fashion mnist +## 1. bigram + + **runing cmd**: `cargo run --package bigram` + + **training result**: accuracy is about 35% for 10 epochs training, quite normal for this simplest language model. + +## 2. fashion mnist + + **runing cmd**: `cargo run --package fashion_mnist` - **runing cmd**: `cargo run --package fashion_mnist` - **current status**: every thing is ok. + **training result**: the model works well. -## 2. yolo v1 +## 3. yolo v1 **runing cmd**: `cargo run --package yolo_v1` - **current status**: The code runs smoothly, but the training process cannot be completed as the loss stops decreasing at around 14, leading to its termination by the burn framework. I'm still investigating the reasons behind this. + **training result**: Training process can't be completed as the loss stops decreasing at around 14, leading to its termination by the burn framework. I'm still investigating the reasons behind this.