Commit 94e0559

authored

Update probabilistic_unet.py

Change beta value to reflect a good starting point and don't get into unstability (derived from #3 )

1 parent 3538c7c commit 94e0559Copy full SHA for 94e0559

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -187,7 +187,7 @@ class ProbabilisticUnet(nn.Module):`
`187`	`187`	`no_cons_per_block: no convs per block in the (convolutional) encoder of prior and posterior`
`188`	`188`	`"""`
`189`	`189`
`190`		`- def __init__(self, input_channels=1, num_classes=1, num_filters=[32,64,128,192], latent_dim=6, no_convs_fcomb=4, beta=1.0):`
	`190`	`+ def __init__(self, input_channels=1, num_classes=1, num_filters=[32,64,128,192], latent_dim=6, no_convs_fcomb=4, beta=10.0):`
`191`	`191`	`super(ProbabilisticUnet, self).__init__()`
`192`	`192`	`self.input_channels = input_channels`
`193`	`193`	`self.num_classes = num_classes`

Comments

(0)