Skip to content

Commit

Permalink
Merge branch 'main' of github.com:richardyang92/burn-examples
Browse files Browse the repository at this point in the history
  • Loading branch information
richardyang92 committed Mar 25, 2024
2 parents 659c6e2 + d845e0c commit 6fe39df
Showing 1 changed file with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
@@ -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.

0 comments on commit 6fe39df

Please sign in to comment.