PyTransformer
PyTransformer copied to clipboard
A visualization and transformation of pytorch model
Hey, i tried to run the demo code in `transform_example.ipynb` and no error occurred. However, when i add `transformer.register(nn.Linear, QLinear)`, there is an error. > TypeError: empty() received an invalid...
I run following script: from PyTransformer.transformers.torchTransformer import TorchTransformer transformer = TorchTransformer() dot = transformer.visualize(self.model, input_tensor = batch['input'], save_name = "/datasets/fig2") And I have "AttributeError: 'tuple' object has no attribute 'clone'...
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()...
https://github.com/ricky40403/PyTransformer/blob/22a0a824be0ef7d4dd65312c4b3e190e4cde4fee/transformers/utils.py#L320 how can I solve the problem?