huggingface / huggingface/candle
Error: cannot find tensor net.b1.0.bn.running_mean
- Dominant language
- Rust
- Stars
- 21k
- Forks
- 1.8k
- Avg merge
- 16h 42m
- Merged PRs (30d)
- 25
Description
When I directly converted the PT file I trained into safetensors, there were no net, head, or fpn as shown in the example
```
model.0.bn.bias
model.0.bn.num_batches_tracked
model.0.bn.running_mean
model.0.bn.running_var
model.0.bn.weight
model.0.conv.weight
model.1.bn.bias
model.1.bn.num_batches_tracked
model.1.bn.running_mean
model.1.bn.running_var
model.1.bn.weight
model.1.conv.weight
model.12.cv1.bn.bias
model.12.cv1.bn.num_batches_tracked
model.12.cv1.bn.running_mean
model.12.cv1.bn.running_var
model.12.cv1.bn.weight
model.12.cv1.conv.weight
model.12.cv2.bn.bias
model.12.cv2.bn.num_batches_tracked
model.12.cv2.bn.running_mean
model.12.cv2.bn.running_var
model.12.cv2.bn.weight
model.12.cv2.conv.weight
model.12.m.0.cv1.bn.bias
model.12.m.0.cv1.bn.num_batches_tracked
model.12.m.0.cv1.bn.running_mean
model.12.m.0.cv1.bn.running_var
model.12.m.0.cv1.bn.weight
model.12.m.0.cv1.conv.weight
model.12.m.0.cv2.bn.bias
```
Contributor guide
No contributing guide indexed for this repository
Research direction
No files, tests, or entry points are named. Start by reproducing the direct PT-to-safetensors conversion and compare its tensor names with the names expected by the failing loader, including net.b1.0.bn.running_mean. Done means the converted model loads without the missing-tensor error.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- pytorch, rust
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100