TypeError: object of type '_ReplaceFunc' has no len()

[sh7jacobi]

I have following error:
if (len(logs) > 1) and (type(logs[0]) == torch.Tensor):
TypeError: object of type '_ReplaceFunc' has no len()
when I run:
from PyTransformer.transformers.torchTransformer import TorchTransformer
transformer = TorchTransformer()
dot = transformer.visualize(self.base_layer, input_tensor = x.detach(), save_name = "/datasets/fig")

[sh7jacobi]

I have the same error for Sequential:
nn.Sequential(
nn.Conv2d(3, 6, kernel_size=7, stride=1,
padding=3, bias=False),
nn.BatchNorm2d(6, momentum=0.1),
nn.ReLU(inplace=True))
I have removed Batch normalization layer and the error disappered. I guess it incorrectly process with some layers, particularly with BatchNorm layer.