Merge pull request #96 from ExponentialML/update/temporal_conv_dropout

Add temporal conv dropout

Author: ExponentialML

models/unet_3d_blocks.py

@@ -276,6 +276,7 @@ def __init__(
             TemporalConvLayer(
                 in_channels,
                 in_channels,
+                dropout=0.1
             )
         ]
         attentions = []
@@ -322,6 +323,7 @@ def __init__(
                 TemporalConvLayer(
                     in_channels,
                     in_channels,
+                    dropout=0.1
                 )
             )
 
@@ -437,6 +439,7 @@ def __init__(
                 TemporalConvLayer(
                     out_channels,
                     out_channels,
+                    dropout=0.1
                 )
             )
             attentions.append(
@@ -575,6 +578,7 @@ def __init__(
                 TemporalConvLayer(
                     out_channels,
                     out_channels,
+                    dropout=0.1
                 )
             )
 
@@ -670,6 +674,7 @@ def __init__(
                 TemporalConvLayer(
                     out_channels,
                     out_channels,
+                    dropout=0.1
                 )
             )
             attentions.append(
@@ -803,6 +808,7 @@ def __init__(
                 TemporalConvLayer(
                     out_channels,
                     out_channels,
+                    dropout=0.1
                 )
             )