Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
MergeVertex with 3D convolution issue #7715
hi, i'm working on 3 u-net model, i'm in beta4 which support 3d convolution, in the following the 3d u-net that i have created:
when i run the code i get the following error from MergeVertex class:
Exception in thread "main" java.lang.IllegalStateException: Unknown input type: InputTypeConvolutional3D(format=NDHWC,d=23,h=32,w=32,c=512)
when i debug in getOutputType in MergeVertex class i get the following values of variables:
Thanks for reporting - fixed here: #7724
You can copy and paste the MergeVertex (configuration) class though and use that instead of the original in your nets.
FYI: The downside of doing that is that if you save and then try to load on a machine without your 'custom' MergeVertex, it'll fail (class not found). For future versions of DL4J though (if your net needs to be usable for that long), you could edit the configuration json in the model zip file to point to the 'real' merge vertex.