Skip to content

Commit 435c595

Browse files
committed
readme: use simpler example
1 parent 63b5ee5 commit 435c595

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,13 @@ model = models.resnet18(weights = "DEFAULT")
1818

1919
topo_loss = TopoLoss(
2020
losses = [
21-
LaplacianPyramid(layer_name = 'fc',factor_h=3.0, factor_w=3.0),
21+
LaplacianPyramid.from_layer(
22+
model=model,
23+
layer = model.fc,
24+
factor_h=3.0,
25+
factor_w=3.0,
26+
scale = 1.0
27+
),
2228
],
2329
)
2430
loss = topo_loss.compute(model=model)

0 commit comments

Comments
 (0)