huggingface / huggingface/candle
Batch training implementation
Open
- Dominant language
- Rust
- Stars
- 21k
- Forks
- 1.8k
- Avg merge
- 16h 42m
- Merged PRs (30d)
- 25
Description
When I train NN with whole data I have cuda out of memory.
Could someone, pls, show the example for batch training?
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by locating Candle's existing neural-network training examples and any APIs related to batching or data loading; the issue names no file or test. Confirm what batch-training support already exists, then define done as a documented example that trains without loading the whole dataset into CUDA memory.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- rust
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100