def forward(self, x, y=None, criterion=None, gpu_nums=1):
    x = self.task(x)
    if not y == None:
        loss = criterion(x, y)
        if gpu_nums > 1：
            loss /= gpu_nums
        loss.backward()
        return x, loss
    return x