Skip to content

Commit 9ad59f1

Browse files
committed
feat: enhance ResNet-finetune model listing output
- Included detailed configuration preview (`cfg:?`) for each model during listing. - Adjusted output format to improve readability by refining weights display structure.
1 parent d87de4d commit 9ad59f1

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
@@ -165,9 +165,12 @@ pub fn train<B: AutodiffBackend>(args: &Args) -> anyhow::Result<()> {
165165
if args.pretrained == "list" {
166166
println!("Available pretrained models:");
167167
for prefab in PREFAB_RESNET_MAP.items {
168+
let cfg = (prefab.builder)();
169+
println!("* \"{}\"", prefab.name);
170+
println!("{cfg:?}");
168171
if let Some(weights) = prefab.weights {
169172
for item in weights.items {
170-
println!("- \"{}.{}\": {}", prefab.name, item.name, item.description);
173+
println!(" - \"{}.{}\"", prefab.name, item.name);
171174
}
172175
}
173176
}

0 commit comments

Comments
 (0)