SuperCarryFu
Results
5
issues of
SuperCarryFu
from torch2trt import * import torch from torchvision.models.alexnet import alexnet if __name__ == '__main__': model = alexnet(pretrained=True).cuda().eval().half() data = torch.randn((1, 3, 224, 224)).cuda().half() data1 = data.contiguous() print('Running torch2trt...') model_trt =...