huggingface / huggingface/candle
Low time effiency when run cnn on mnist-traning only with CPU
- Dominant language
- Rust
- Stars
- 21k
- Forks
- 1.8k
- Avg merge
- 16h 42m
- Merged PRs (30d)
- 25
Description
I try to use the following command to train the CNN model with mnist-training example. It works well when training linear and mlp model, but when I config it to train the CNN model, it needs approximately 1hr30mins per epoch. I think it may be abnormal.
`cargo run --example mnist-training --features="candle-datasets" cnn --epochs=5 --save=cnn.safetensors`
Can you please try it?
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by running the reported `cargo run --example mnist-training --features="candle-datasets" cnn --epochs=5 --save=cnn.safetensors` command on CPU and compare its epoch time with the linear and MLP modes. Done means identifying whether the CNN timing is abnormal and documenting or addressing the cause, with reproducible timing details.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- rust
- Domain
- machine-learning, performance
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100