显存与数据大小的关系

研究显存与数据大小的关系的意义在于，我们能够确定自己的网络在给定的显卡上是否跑得起来。
以下是一个测试程序，用来测试数据占用显存的。

import math
import torch
import numpy as np
import torch.nn.functional as F


def gradient_loss(s, penalty='l2')<