if is_layer(layer, "Pooling"):
return MaxPooling2D(padding="same")
if is_layer(layer, "BatchNormalization"):
return BatchNormalization()
if is_layer(layer, "Concatenate"):
return Concatenate()
if is_layer(layer, "Add"):
return Add()
After Change
if is_layer(layer, "ReLU"):
return torch.nn.ReLU()
if is_layer(layer, "Softmax"):
return torch.nn.Softmax()
if is_layer(layer, "Flatten"):
return TorchFlatten()