Commit 8015281

author

Stefan Knegt

authored

Merge pull request #4 from justusschock/patch-1

Update default parameters

2 parents 3538c7c + 94e0559 commit 8015281Copy full SHA for 8015281

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)