huggingface / huggingface/candle

Low time effiency when run cnn on mnist-traning only with CPU

Open
#2,144 3 comments 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.