import numpy as np import torch import torch.nn as nn import torch.nn.functional as F import math _weights_dict = dict() def load_weights(weight_file): if weight_file == None: return try: weights_dict = np.load(weight_file, allow_pickle=True).item() except: weights_dict = np.load(weight_file, allow_pickle=True, encoding='bytes').item() return weights_dict class KitModel(nn.Module): def __init__(self, weight_file): super(KitModel, self).__init__() global _weights_dict _weights_dict = load_weights(weight_file) self.InceptionV3_InceptionV3_Conv2d_1a_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Conv2d_1a_3x3/Conv2D', in_channels=3, out_channels=32, kernel_size=(3, 3), stride=(2, 2), groups=1, bias=None) self.InceptionV3_InceptionV3_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Conv2d_1a_3x3/BatchNorm/FusedBatchNorm', num_features=32, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Conv2d_2a_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Conv2d_2a_3x3/Conv2D', in_channels=32, out_channels=32, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Conv2d_2a_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Conv2d_2a_3x3/BatchNorm/FusedBatchNorm', num_features=32, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Conv2d_2b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Conv2d_2b_3x3/Conv2D', in_channels=32, out_channels=64, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Conv2d_2b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Conv2d_2b_3x3/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Conv2d_3b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Conv2d_3b_1x1/Conv2D', in_channels=64, out_channels=80, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Conv2d_3b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Conv2d_3b_1x1/BatchNorm/FusedBatchNorm', num_features=80, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Conv2d_4a_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Conv2d_4a_3x3/Conv2D', in_channels=80, out_channels=192, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Conv2d_4a_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Conv2d_4a_3x3/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=192, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=192, out_channels=48, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=192, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=48, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=192, out_channels=32, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=32, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_1/Conv2d_0b_5x5/Conv2D', in_channels=48, out_channels=64, kernel_size=(5, 5), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_2/Conv2d_0b_3x3/Conv2D', in_channels=64, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_1/Conv2d_0b_5x5/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_2/Conv2d_0b_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5b/Branch_2/Conv2d_0c_3x3/Conv2D', in_channels=96, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5b/Branch_2/Conv2d_0c_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=256, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_1/Conv2d_0b_1x1/Conv2D', in_channels=256, out_channels=48, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=256, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_1/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=48, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=256, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_1/Conv_1_0c_5x5/Conv2D', in_channels=48, out_channels=64, kernel_size=(5, 5), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_2/Conv2d_0b_3x3/Conv2D', in_channels=64, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_1/Conv_1_0c_5x5/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_2/Conv2d_0b_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5c/Branch_2/Conv2d_0c_3x3/Conv2D', in_channels=96, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5c/Branch_2/Conv2d_0c_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=288, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=288, out_channels=48, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=288, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=48, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=288, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_1/Conv2d_0b_5x5/Conv2D', in_channels=48, out_channels=64, kernel_size=(5, 5), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_2/Conv2d_0b_3x3/Conv2D', in_channels=64, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_1/Conv2d_0b_5x5/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_2/Conv2d_0b_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_5d/Branch_2/Conv2d_0c_3x3/Conv2D', in_channels=96, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_5d/Branch_2/Conv2d_0c_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6a/Branch_0/Conv2d_1a_1x1/Conv2D', in_channels=288, out_channels=384, kernel_size=(3, 3), stride=(2, 2), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6a/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=288, out_channels=64, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6a/Branch_0/Conv2d_1a_1x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6a/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=64, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6a/Branch_1/Conv2d_0b_3x3/Conv2D', in_channels=64, out_channels=96, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6a/Branch_1/Conv2d_0b_3x3/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6a/Branch_1/Conv2d_1a_1x1/Conv2D', in_channels=96, out_channels=96, kernel_size=(3, 3), stride=(2, 2), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6a/Branch_1/Conv2d_1a_1x1/BatchNorm/FusedBatchNorm', num_features=96, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=128, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=128, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_1/Conv2d_0b_1x7/Conv2D', in_channels=128, out_channels=128, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0b_7x1/Conv2D', in_channels=128, out_channels=128, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_1/Conv2d_0b_1x7/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0b_7x1/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_1/Conv2d_0c_7x1/Conv2D', in_channels=128, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0c_1x7/Conv2D', in_channels=128, out_channels=128, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_1/Conv2d_0c_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0c_1x7/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0d_7x1/Conv2D', in_channels=128, out_channels=128, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0d_7x1/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0e_1x7/Conv2D', in_channels=128, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6b/Branch_2/Conv2d_0e_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=160, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=160, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_1/Conv2d_0b_1x7/Conv2D', in_channels=160, out_channels=160, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0b_7x1/Conv2D', in_channels=160, out_channels=160, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_1/Conv2d_0b_1x7/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0b_7x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_1/Conv2d_0c_7x1/Conv2D', in_channels=160, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0c_1x7/Conv2D', in_channels=160, out_channels=160, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_1/Conv2d_0c_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0c_1x7/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0d_7x1/Conv2D', in_channels=160, out_channels=160, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0d_7x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0e_1x7/Conv2D', in_channels=160, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6c/Branch_2/Conv2d_0e_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=160, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=160, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_1/Conv2d_0b_1x7/Conv2D', in_channels=160, out_channels=160, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0b_7x1/Conv2D', in_channels=160, out_channels=160, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_1/Conv2d_0b_1x7/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0b_7x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_1/Conv2d_0c_7x1/Conv2D', in_channels=160, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0c_1x7/Conv2D', in_channels=160, out_channels=160, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_1/Conv2d_0c_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0c_1x7/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0d_7x1/Conv2D', in_channels=160, out_channels=160, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0d_7x1/BatchNorm/FusedBatchNorm', num_features=160, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0e_1x7/Conv2D', in_channels=160, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6d/Branch_2/Conv2d_0e_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_1/Conv2d_0b_1x7/Conv2D', in_channels=192, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0b_7x1/Conv2D', in_channels=192, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_1/Conv2d_0b_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0b_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_1/Conv2d_0c_7x1/Conv2D', in_channels=192, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0c_1x7/Conv2D', in_channels=192, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_1/Conv2d_0c_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0c_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0d_7x1/Conv2D', in_channels=192, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0d_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0e_1x7/Conv2D', in_channels=192, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_6e/Branch_2/Conv2d_0e_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7a/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=768, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7a/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_AuxLogits_Conv2d_1b_1x1_Conv2D = self.__conv(2, name='InceptionV3/AuxLogits/Conv2d_1b_1x1/Conv2D', in_channels=768, out_channels=128, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_AuxLogits_Conv2d_1b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/AuxLogits/Conv2d_1b_1x1/BatchNorm/FusedBatchNorm', num_features=128, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7a/Branch_0/Conv2d_1a_3x3/Conv2D', in_channels=192, out_channels=320, kernel_size=(3, 3), stride=(2, 2), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_0b_1x7/Conv2D', in_channels=192, out_channels=192, kernel_size=(1, 7), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7a/Branch_0/Conv2d_1a_3x3/BatchNorm/FusedBatchNorm', num_features=320, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_0b_1x7/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_AuxLogits_Conv2d_2a_5x5_Conv2D = self.__conv(2, name='InceptionV3/AuxLogits/Conv2d_2a_5x5/Conv2D', in_channels=128, out_channels=768, kernel_size=(5, 5), stride=(1, 1), groups=1, bias=None) self.InceptionV3_AuxLogits_Conv2d_2a_5x5_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/AuxLogits/Conv2d_2a_5x5/BatchNorm/FusedBatchNorm', num_features=768, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_0c_7x1/Conv2D', in_channels=192, out_channels=192, kernel_size=(7, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_0c_7x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_AuxLogits_Conv2d_2b_1x1_Conv2D = self.__conv(2, name='InceptionV3/AuxLogits/Conv2d_2b_1x1/Conv2D', in_channels=768, out_channels=1001, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=True) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_1a_3x3/Conv2D', in_channels=192, out_channels=192, kernel_size=(3, 3), stride=(2, 2), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7a/Branch_1/Conv2d_1a_3x3/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=1280, out_channels=320, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=1280, out_channels=384, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=1280, out_channels=448, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=320, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=448, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=1280, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_1/Conv2d_0b_1x3/Conv2D', in_channels=384, out_channels=384, kernel_size=(1, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_1/Conv2d_0b_3x1/Conv2D', in_channels=384, out_channels=384, kernel_size=(3, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0b_3x3/Conv2D', in_channels=448, out_channels=384, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_1/Conv2d_0b_1x3/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_1/Conv2d_0b_3x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0b_3x3/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0c_1x3/Conv2D', in_channels=384, out_channels=384, kernel_size=(1, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0d_3x1/Conv2D', in_channels=384, out_channels=384, kernel_size=(3, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0c_1x3/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7b/Branch_2/Conv2d_0d_3x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_0/Conv2d_0a_1x1/Conv2D', in_channels=2048, out_channels=320, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_1/Conv2d_0a_1x1/Conv2D', in_channels=2048, out_channels=384, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0a_1x1/Conv2D', in_channels=2048, out_channels=448, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_0/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=320, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_1/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0a_1x1/BatchNorm/FusedBatchNorm', num_features=448, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/Conv2D', in_channels=2048, out_channels=192, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/BatchNorm/FusedBatchNorm', num_features=192, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_1/Conv2d_0b_1x3/Conv2D', in_channels=384, out_channels=384, kernel_size=(1, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_1/Conv2d_0c_3x1/Conv2D', in_channels=384, out_channels=384, kernel_size=(3, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0b_3x3/Conv2D', in_channels=448, out_channels=384, kernel_size=(3, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_1/Conv2d_0b_1x3/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_1/Conv2d_0c_3x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0b_3x3/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0c_1x3/Conv2D', in_channels=384, out_channels=384, kernel_size=(1, 3), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Conv2D = self.__conv(2, name='InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0d_3x1/Conv2D', in_channels=384, out_channels=384, kernel_size=(3, 1), stride=(1, 1), groups=1, bias=None) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0c_1x3/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm = self.__batch_normalization(2, 'InceptionV3/InceptionV3/Mixed_7c/Branch_2/Conv2d_0d_3x1/BatchNorm/FusedBatchNorm', num_features=384, eps=0.0010000000474974513, momentum=0.0) self.InceptionV3_Logits_Conv2d_1c_1x1_Conv2D = self.__conv(2, name='InceptionV3/Logits/Conv2d_1c_1x1/Conv2D', in_channels=2048, out_channels=1001, kernel_size=(1, 1), stride=(1, 1), groups=1, bias=True) def forward(self, x): InceptionV3_InceptionV3_Conv2d_1a_3x3_Conv2D = self.InceptionV3_InceptionV3_Conv2d_1a_3x3_Conv2D(x) InceptionV3_InceptionV3_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Conv2d_1a_3x3_Conv2D) InceptionV3_InceptionV3_Conv2d_1a_3x3_Relu = F.relu(InceptionV3_InceptionV3_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Conv2d_2a_3x3_Conv2D = self.InceptionV3_InceptionV3_Conv2d_2a_3x3_Conv2D(InceptionV3_InceptionV3_Conv2d_1a_3x3_Relu) InceptionV3_InceptionV3_Conv2d_2a_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Conv2d_2a_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Conv2d_2a_3x3_Conv2D) InceptionV3_InceptionV3_Conv2d_2a_3x3_Relu = F.relu(InceptionV3_InceptionV3_Conv2d_2a_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Conv2d_2b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Conv2d_2a_3x3_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Conv2d_2b_3x3_Conv2D = self.InceptionV3_InceptionV3_Conv2d_2b_3x3_Conv2D(InceptionV3_InceptionV3_Conv2d_2b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Conv2d_2b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Conv2d_2b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Conv2d_2b_3x3_Conv2D) InceptionV3_InceptionV3_Conv2d_2b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Conv2d_2b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_MaxPool_3a_3x3_MaxPool, InceptionV3_InceptionV3_MaxPool_3a_3x3_MaxPool_idx = F.max_pool2d(InceptionV3_InceptionV3_Conv2d_2b_3x3_Relu, kernel_size=(3, 3), stride=(2, 2), padding=0, ceil_mode=False, return_indices=True) InceptionV3_InceptionV3_Conv2d_3b_1x1_Conv2D = self.InceptionV3_InceptionV3_Conv2d_3b_1x1_Conv2D(InceptionV3_InceptionV3_MaxPool_3a_3x3_MaxPool) InceptionV3_InceptionV3_Conv2d_3b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Conv2d_3b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Conv2d_3b_1x1_Conv2D) InceptionV3_InceptionV3_Conv2d_3b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Conv2d_3b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Conv2d_4a_3x3_Conv2D = self.InceptionV3_InceptionV3_Conv2d_4a_3x3_Conv2D(InceptionV3_InceptionV3_Conv2d_3b_1x1_Relu) InceptionV3_InceptionV3_Conv2d_4a_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Conv2d_4a_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Conv2d_4a_3x3_Conv2D) InceptionV3_InceptionV3_Conv2d_4a_3x3_Relu = F.relu(InceptionV3_InceptionV3_Conv2d_4a_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_MaxPool_5a_3x3_MaxPool, InceptionV3_InceptionV3_MaxPool_5a_3x3_MaxPool_idx = F.max_pool2d(InceptionV3_InceptionV3_Conv2d_4a_3x3_Relu, kernel_size=(3, 3), stride=(2, 2), padding=0, ceil_mode=False, return_indices=True) InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_MaxPool_5a_3x3_MaxPool) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_MaxPool_5a_3x3_MaxPool) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_MaxPool_5a_3x3_MaxPool) InceptionV3_InceptionV3_Mixed_5b_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_MaxPool_5a_3x3_MaxPool, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5b_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0a_1x1_Relu, (2, 2, 2, 2)) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Conv2D(InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0a_1x1_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0b_3x3_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5b_concat = torch.cat((InceptionV3_InceptionV3_Mixed_5b_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_5b_Branch_1_Conv2d_0b_5x5_Relu, InceptionV3_InceptionV3_Mixed_5b_Branch_2_Conv2d_0c_3x3_Relu, InceptionV3_InceptionV3_Mixed_5b_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5b_concat) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5b_concat) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5b_concat) InceptionV3_InceptionV3_Mixed_5c_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_5b_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5c_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv2d_0b_1x1_Relu, (2, 2, 2, 2)) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Conv2D(InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0a_1x1_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0b_3x3_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5c_concat = torch.cat((InceptionV3_InceptionV3_Mixed_5c_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_5c_Branch_1_Conv_1_0c_5x5_Relu, InceptionV3_InceptionV3_Mixed_5c_Branch_2_Conv2d_0c_3x3_Relu, InceptionV3_InceptionV3_Mixed_5c_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5c_concat) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5c_concat) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5c_concat) InceptionV3_InceptionV3_Mixed_5d_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_5c_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5d_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0a_1x1_Relu, (2, 2, 2, 2)) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Conv2D(InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0a_1x1_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0b_3x3_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_5d_concat = torch.cat((InceptionV3_InceptionV3_Mixed_5d_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_5d_Branch_1_Conv2d_0b_5x5_Relu, InceptionV3_InceptionV3_Mixed_5d_Branch_2_Conv2d_0c_3x3_Relu, InceptionV3_InceptionV3_Mixed_5d_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5d_concat) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_5d_concat) InceptionV3_InceptionV3_Mixed_6a_Branch_2_MaxPool_1a_3x3_MaxPool, InceptionV3_InceptionV3_Mixed_6a_Branch_2_MaxPool_1a_3x3_MaxPool_idx = F.max_pool2d(InceptionV3_InceptionV3_Mixed_5d_concat, kernel_size=(3, 3), stride=(2, 2), padding=0, ceil_mode=False, return_indices=True) InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0a_1x1_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_0b_3x3_Relu) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6a_concat = torch.cat((InceptionV3_InceptionV3_Mixed_6a_Branch_0_Conv2d_1a_1x1_Relu, InceptionV3_InceptionV3_Mixed_6a_Branch_1_Conv2d_1a_1x1_Relu, InceptionV3_InceptionV3_Mixed_6a_Branch_2_MaxPool_1a_3x3_MaxPool,), 1) InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6a_concat) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6a_concat) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6a_concat) InceptionV3_InceptionV3_Mixed_6b_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_6a_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0a_1x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0a_1x1_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0b_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0b_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0c_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0d_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6b_concat = torch.cat((InceptionV3_InceptionV3_Mixed_6b_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_6b_Branch_1_Conv2d_0c_7x1_Relu, InceptionV3_InceptionV3_Mixed_6b_Branch_2_Conv2d_0e_1x7_Relu, InceptionV3_InceptionV3_Mixed_6b_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_concat) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_concat) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6b_concat) InceptionV3_InceptionV3_Mixed_6c_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_6b_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0a_1x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0a_1x1_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0b_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0b_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0c_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0d_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6c_concat = torch.cat((InceptionV3_InceptionV3_Mixed_6c_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_6c_Branch_1_Conv2d_0c_7x1_Relu, InceptionV3_InceptionV3_Mixed_6c_Branch_2_Conv2d_0e_1x7_Relu, InceptionV3_InceptionV3_Mixed_6c_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_concat) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_concat) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6c_concat) InceptionV3_InceptionV3_Mixed_6d_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_6c_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0a_1x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0a_1x1_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0b_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0b_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0c_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0d_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6d_concat = torch.cat((InceptionV3_InceptionV3_Mixed_6d_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_6d_Branch_1_Conv2d_0c_7x1_Relu, InceptionV3_InceptionV3_Mixed_6d_Branch_2_Conv2d_0e_1x7_Relu, InceptionV3_InceptionV3_Mixed_6d_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_concat) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_concat) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6d_concat) InceptionV3_InceptionV3_Mixed_6e_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_6d_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0a_1x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0a_1x1_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0b_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0b_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0c_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0d_7x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Conv2D_pad) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Conv2D) InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_6e_concat = torch.cat((InceptionV3_InceptionV3_Mixed_6e_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_6e_Branch_1_Conv2d_0c_7x1_Relu, InceptionV3_InceptionV3_Mixed_6e_Branch_2_Conv2d_0e_1x7_Relu, InceptionV3_InceptionV3_Mixed_6e_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6e_concat) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_6e_concat) InceptionV3_InceptionV3_Mixed_7a_Branch_2_MaxPool_1a_3x3_MaxPool, InceptionV3_InceptionV3_Mixed_7a_Branch_2_MaxPool_1a_3x3_MaxPool_idx = F.max_pool2d(InceptionV3_InceptionV3_Mixed_6e_concat, kernel_size=(3, 3), stride=(2, 2), padding=0, ceil_mode=False, return_indices=True) InceptionV3_AuxLogits_AvgPool_1a_5x5_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_6e_concat, kernel_size=(5, 5), stride=(3, 3), padding=(0,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_AuxLogits_Conv2d_1b_1x1_Conv2D = self.InceptionV3_AuxLogits_Conv2d_1b_1x1_Conv2D(InceptionV3_AuxLogits_AvgPool_1a_5x5_AvgPool) InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_AuxLogits_Conv2d_1b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_AuxLogits_Conv2d_1b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_AuxLogits_Conv2d_1b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_0a_1x1_Relu) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0a_1x1_Relu, (3, 3, 0, 0)) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Conv2D = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Conv2D(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Conv2D_pad) InceptionV3_AuxLogits_Conv2d_1b_1x1_Relu = F.relu(InceptionV3_AuxLogits_Conv2d_1b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Conv2D) InceptionV3_AuxLogits_Conv2d_2a_5x5_Conv2D = self.InceptionV3_AuxLogits_Conv2d_2a_5x5_Conv2D(InceptionV3_AuxLogits_Conv2d_1b_1x1_Relu) InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_BatchNorm_FusedBatchNorm) InceptionV3_AuxLogits_Conv2d_2a_5x5_BatchNorm_FusedBatchNorm = self.InceptionV3_AuxLogits_Conv2d_2a_5x5_BatchNorm_FusedBatchNorm(InceptionV3_AuxLogits_Conv2d_2a_5x5_Conv2D) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0b_1x7_Relu, (0, 0, 3, 3)) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Conv2D(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Conv2D_pad) InceptionV3_AuxLogits_Conv2d_2a_5x5_Relu = F.relu(InceptionV3_AuxLogits_Conv2d_2a_5x5_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Conv2D) InceptionV3_AuxLogits_Conv2d_2b_1x1_Conv2D = self.InceptionV3_AuxLogits_Conv2d_2b_1x1_Conv2D(InceptionV3_AuxLogits_Conv2d_2a_5x5_Relu) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_BatchNorm_FusedBatchNorm) InceptionV3_AuxLogits_SpatialSqueeze = torch.squeeze(InceptionV3_AuxLogits_Conv2d_2b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_0c_7x1_Relu) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7a_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7a_Branch_0_Conv2d_1a_3x3_Relu, InceptionV3_InceptionV3_Mixed_7a_Branch_1_Conv2d_1a_3x3_Relu, InceptionV3_InceptionV3_Mixed_7a_Branch_2_MaxPool_1a_3x3_MaxPool,), 1) InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7a_concat) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7a_concat) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7a_concat) InceptionV3_InceptionV3_Mixed_7b_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_7a_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7b_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Relu, (1, 1, 0, 0)) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Conv2D(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0a_1x1_Relu, (0, 0, 1, 1)) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Conv2D(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0a_1x1_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_1_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_1x3_Relu, InceptionV3_InceptionV3_Mixed_7b_Branch_1_Conv2d_0b_3x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Relu, (1, 1, 0, 0)) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Conv2D(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0b_3x3_Relu, (0, 0, 1, 1)) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Conv2D(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Conv2D) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7b_Branch_2_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0c_1x3_Relu, InceptionV3_InceptionV3_Mixed_7b_Branch_2_Conv2d_0d_3x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_7b_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7b_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_7b_Branch_1_concat, InceptionV3_InceptionV3_Mixed_7b_Branch_2_concat, InceptionV3_InceptionV3_Mixed_7b_Branch_3_Conv2d_0b_1x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7b_concat) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7b_concat) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7b_concat) InceptionV3_InceptionV3_Mixed_7c_Branch_3_AvgPool_0a_3x3_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_7b_concat, kernel_size=(3, 3), stride=(1, 1), padding=(1,), ceil_mode=False, count_include_pad=False) InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_Conv2D(InceptionV3_InceptionV3_Mixed_7c_Branch_3_AvgPool_0a_3x3_AvgPool) InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Relu, (1, 1, 0, 0)) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Conv2D(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0a_1x1_Relu, (0, 0, 1, 1)) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Conv2D(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0a_1x1_Relu, (1, 1, 1, 1)) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Conv2D(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_1_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0b_1x3_Relu, InceptionV3_InceptionV3_Mixed_7c_Branch_1_Conv2d_0c_3x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Relu, (1, 1, 0, 0)) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Conv2D(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Conv2D_pad = F.pad(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0b_3x3_Relu, (0, 0, 1, 1)) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Conv2D = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Conv2D(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Conv2D_pad) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm = self.InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Conv2D) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Relu = F.relu(InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_BatchNorm_FusedBatchNorm) InceptionV3_InceptionV3_Mixed_7c_Branch_2_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0c_1x3_Relu, InceptionV3_InceptionV3_Mixed_7c_Branch_2_Conv2d_0d_3x1_Relu,), 1) InceptionV3_InceptionV3_Mixed_7c_concat = torch.cat((InceptionV3_InceptionV3_Mixed_7c_Branch_0_Conv2d_0a_1x1_Relu, InceptionV3_InceptionV3_Mixed_7c_Branch_1_concat, InceptionV3_InceptionV3_Mixed_7c_Branch_2_concat, InceptionV3_InceptionV3_Mixed_7c_Branch_3_Conv2d_0b_1x1_Relu,), 1) kernel_size = self._reduced_kernel_size_for_small_input(InceptionV3_InceptionV3_Mixed_7c_concat, [8,8]) InceptionV3_Logits_AvgPool_1a_8x8_AvgPool = F.avg_pool2d(InceptionV3_InceptionV3_Mixed_7c_concat, kernel_size=(kernel_size[0], kernel_size[1]), stride=(2, 2), padding=(0,), ceil_mode=False, count_include_pad=False) InceptionV3_Logits_Conv2d_1c_1x1_Conv2D = self.InceptionV3_Logits_Conv2d_1c_1x1_Conv2D(InceptionV3_Logits_AvgPool_1a_8x8_AvgPool) InceptionV3_Logits_SpatialSqueeze = torch.squeeze(InceptionV3_Logits_Conv2d_1c_1x1_Conv2D) MMdnn_Output_input = [InceptionV3_Logits_SpatialSqueeze,InceptionV3_AuxLogits_SpatialSqueeze] return MMdnn_Output_input def _reduced_kernel_size_for_small_input(self, input_tensor, kernel_size): shape = input_tensor.shape if shape[2] is None or shape[3] is None: kernel_size_out = kernel_size else: kernel_size_out = [min(shape[2], kernel_size[0]), min(shape[3], kernel_size[1])] return kernel_size_out @staticmethod def __batch_normalization(dim, name, **kwargs): if dim == 0 or dim == 1: layer = nn.BatchNorm1d(**kwargs) elif dim == 2: layer = nn.BatchNorm2d(**kwargs) elif dim == 3: layer = nn.BatchNorm3d(**kwargs) else: raise NotImplementedError() if 'scale' in _weights_dict[name]: layer.state_dict()['weight'].copy_(torch.from_numpy(_weights_dict[name]['scale'])) else: layer.weight.data.fill_(1) if 'bias' in _weights_dict[name]: layer.state_dict()['bias'].copy_(torch.from_numpy(_weights_dict[name]['bias'])) else: layer.bias.data.fill_(0) layer.state_dict()['running_mean'].copy_(torch.from_numpy(_weights_dict[name]['mean'])) layer.state_dict()['running_var'].copy_(torch.from_numpy(_weights_dict[name]['var'])) return layer @staticmethod def __conv(dim, name, **kwargs): if dim == 1: layer = nn.Conv1d(**kwargs) elif dim == 2: layer = nn.Conv2d(**kwargs) elif dim == 3: layer = nn.Conv3d(**kwargs) else: raise NotImplementedError() layer.state_dict()['weight'].copy_(torch.from_numpy(_weights_dict[name]['weights'])) if 'bias' in _weights_dict[name]: layer.state_dict()['bias'].copy_(torch.from_numpy(_weights_dict[name]['bias'])) return layer