From 57e661040ae73c565f568bbbf8a4d27c4c04537a Mon Sep 17 00:00:00 2001
From: "Mikkel N. Schmidt" <mnsc@dtu.dk>
Date: Wed, 27 Apr 2022 09:55:20 +0200
Subject: [PATCH] demo updated with map estimate
---
demos/two_moons.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/demos/two_moons.py b/demos/two_moons.py
index 865896f..95716d5 100644
--- a/demos/two_moons.py
+++ b/demos/two_moons.py
@@ -26,7 +26,7 @@ variables = 2
channels = 4
tracks = 4
network = nn.Sequential(
- NormalLeaf(tracks, variables, channels),
+ NormalLeaf(tracks, variables, channels, n=N, alpha0=1, beta0=1),
Weightsum(tracks, 1, channels)
)
print(f"Total parameters: {sum(p.numel() for p in network.parameters() if p.requires_grad)}")
--
GitLab