MauroPfister

import torch
from ops.dcn import DeformConvPack  # Needs dcn module from https://github.com/open-mmlab/mmdetection

torch.manual_seed(0)


def check_grad():

    device = torch.device('cuda')

MauroPfister

import torch
from torch import nn
import torchvision
from ops.dcn import DeformConv
import time


class NormalConv(nn.Module):

    def __init__(self, in_channels, groups):