1 2 3 4 5 6 7
| device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model = SuperPointNet_gauss2() model = model.to(device)
# check keras-like model summary using torchsummary from torchsummary import summary summary(model, input_size=(1, 240, 320))
|