microsoft
/
EdgeML
зеркало из https://github.com/microsoft/EdgeML.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

model size call works

This commit is contained in:
Ubuntu 2019-08-15 16:10:16 +00:00
Родитель d373338570
Коммит 4db7109529
1 изменённых файлов: 3 добавлений и 3 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

6
edgeml/pytorch/graph/rnn.py
Просмотреть файл

@ -136,11 +136,11 @@ class RNNCell(nn.Module):
totalnnz = 2 # For Zeta and Nu
for i in range(0, endW):
totalnnz += utils.countNNZ(mats[i], self._wSparsity)
totalnnz += utils.countNNZ(mats[i].cpu(), self._wSparsity)
for i in range(endW, endU):
totalnnz += utils.countNNZ(mats[i], self._uSparsity)
totalnnz += utils.countNNZ(mats[i].cpu(), self._uSparsity)
for i in range(endU, len(mats)):
totalnnz += utils.countNNZ(mats[i], False)
totalnnz += utils.countNNZ(mats[i].cpu(), False)
return totalnnz * 4
def copy_previous_UW(self):