У меня внутри функции такое идет: with torch.no_grad():

Question

У меня внутри функции такое идет: with torch.no_grad():

net = NeuralNetwork()
if cuda:
torch.cuda.empty_cache()
net.load_state_dict(copyStateDict(torch.load(trained_model)))
else:
net.load_state_dict(copyStateDict(torch.load(trained_model, map_location='cpu')))
Получается, нужно вынести net=NeuralNetwork() из torch.no_grad , настроить автоматическую загрузку весов при инициализации и передавать net уже с загруженными весами внутрь функции для всех инференций, верно?

#backend #programming #python #russian

0

13.09.2019