Пишу везде где можно cuda: nn.Linear(c7, h1).cuda() torch.Tensor(...).cuda() model = model.cuda() даже на это

Question

Пишу везде где можно cuda: nn.Linear(c7, h1).cuda() torch.Tensor(...).cuda() model = model.cuda() даже на это

не поругалось:

nn.MSELoss().cuda()
datas = torch.sigmoid(self.hid2(datas)).cuda()

Полагаю, что дольше из-за того, что перекопируется между видеопаматью и обычной, а где? циклы и функции заставляют копировать в cpu а потом обратно?

0

12.08.2018

1 ответов

8 просмотров

Evgenii Zheltonozhskii🇮🇱 · Accepted Answer

Evgenii Zheltonozhskii🇮🇱

теперь уже не .cuda() а .to()

0

12.08.2018

Похожие чаты

Пишу везде где можно cuda: nn.Linear(c7, h1).cuda() torch.Tensor(...).cuda() model = model.cuda() даже на это

1 ответов

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