Fix readme

aleximmer · Mar 14, 2024 · 30e28f2 · 30e28f2
1 parent 894f640
commit 30e28f2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -69,7 +69,9 @@ pred = la(x, link_approx='probit')
 torch.save(la.state_dict(), 'state_dict.bin')
 
 # Load serialized Laplace
-la2 = Laplace(model, 'regression', subset_of_weights='all', hessian_structure='full')
+la2 = Laplace(model, 'classification',
+              subset_of_weights='all',
+              hessian_structure='diag')
 la2.load_state_dict(torch.load('state_dict.bin'))
 ```