Skip to content

Commit 297337a

Browse files
committed
feat: enhance weights display in ResNet-finetune model listing
- Added item descriptions to weights display for improved clarity.
1 parent 9ad59f1 commit 297337a

File tree

1 file changed

+4
-1
lines changed
  • examples/resnet-finetune/src

1 file changed

+4
-1
lines changed

examples/resnet-finetune/src/main.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,10 @@ pub fn train<B: AutodiffBackend>(args: &Args) -> anyhow::Result<()> {
170170
println!("{cfg:?}");
171171
if let Some(weights) = prefab.weights {
172172
for item in weights.items {
173-
println!(" - \"{}.{}\"", prefab.name, item.name);
173+
println!(
174+
" - \"{}.{}\": {}",
175+
prefab.name, item.name, item.description
176+
);
174177
}
175178
}
176179
}

0 commit comments

Comments
 (0)