X = torch.zeros((32, 32, 128, 128)).cuda(); conv = torch.nn.Conv2d(32, 64, 1,

Question

X = torch.zeros((32, 32, 128, 128)).cuda(); conv = torch.nn.Conv2d(32, 64, 1,

1, padding=1, bias=False).cuda();

with torch.no_grad():
start = time.time();
for _ in range(10):
y = conv(x);
torch.cuda.synchronize();
print(time.time() - start);

Интересно, где ему тут в память упереться?

0

24.08.2021

2 ответов

38 просмотров

Лис Электронщик

16МБ — должно влезать в кеш l2 карты полностью

0

24.08.2021

Evgenii Zheltonozhskii🇮🇱 · Accepted Answer

Evgenii Zheltonozhskii🇮🇱

Так наоборот

0

24.08.2021

Похожие чаты

X = torch.zeros((32, 32, 128, 128)).cuda(); conv = torch.nn.Conv2d(32, 64, 1,

2 ответов

Похожие вопросы