Buster

you can try it ``` model = torch.load('./mbv3_small.pth', map_location = 'cpu') print('Loading base network...') weight = model["state_dict"] base_net = torch.nn.DataParallel(net) base_net.load_state_dict(weight) y = net(x) end = time.time() print('Running time: %s...