Skip to content
Discussion options

You must be logged in to vote

Hi,

setting spatial_dims=3 is correct in this case. dropout_dim and norm_dim depend on your architecture. I would advice starting with copying prominent architectures like the ResNet for classification/regression and the UNet for segmentation tasks.

Unstable training can have many reasons. Make sure to start with a small simple network and gradually make it more complex. A Recipe for Training Neural Networks by Andrej Karpathy is probably the best starting point I know. Good look! :)

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by n-faro
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants