TensorFlow-Advanced-Segmentation-Models

TensorFlow-Advanced-Segmentation-Models copied to clipboard

Is there exist performance issue when training or forward model pass? take UNet for example， ··· def call(self, inputs, training=None, mask=None): ···

when training or prediction， self.backbone(inputs) is calculated for 5 times, but the input and backbone not changed，so can this code can be changed to x0, x1, x2, x3, x4 = self.backbone(inputs, training=training) self.upsample2d_x2_block function can use x0, x1, x2, x3, x4 , in this way, the backbone will calculate only 1 time.

thank U.