No module named 'torch2trt.module_test'

import torch2trt works fine, but from torch2trt import torch2trt would raise an error:

torch2trt/torch2trt/converters/activation.py in <module>()
      1 from torch2trt.torch2trt import *
----> 2 from torch2trt.module_test import add_module_test
      3 from .unary import UnaryModule
      4 
      5 
ModuleNotFoundError: No module named 'torch2trt.module_test'

Since import torch2trt just works, I also tried torch2trt.torch2trt(...) instead of torch2trt(...), but it still raises an error: AttributeError: module 'torch2trt' has no attribute 'torch2trt'.

[sebastian-sz]

To install the torch2trt you can run python setup.py install after cloning the repo. After that you should be able to from torch2trt import torch2trt

[zifuwanggg]

I had the above issue when I used some un-verified commit (I forgot which one). I think the problem is already solved.