研究显存与数据大小的关系的意义在于，我们能够确定自己的网络在给定的显卡上是否跑得起来。
以下是一个测试程序，用来测试数据占用显存的。
import math
import torch
import numpy as np
import torch.nn.functional as F
def gradient_loss(s, penalty='l2'):
dy = torch.abs显存与数据大小的关系
猜你喜欢
转载自blog.csdn.net/juluwangriyue/article/details/120914482
今日推荐
周排行