yannadani / dlgm Goto Github PK

S = v.view(n,m)
for i in range(N):
        S = torch.mm(S, torch.mm(torch.ones(1,n),S).inverse())
        S = torch.mm(torch.mv(S, torch.ones(m,1)).inverse(), S)
return S

log path logs/sintel/sintel_seed-7.log
Using 1 GPUs!
Traceback (most recent call last):
  File "main.py", line 294, in <module>
    model = model.to(args.device)
  File "/home/manan/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 386, in to
    return self._apply(convert)
  File "/home/manan/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 193, in _apply
    module._apply(fn)
  File "/home/manan/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 199, in _apply
    param.data = fn(param.data)
  File "/home/manan/.local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 384, in convert
    return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
RuntimeError: CUDA error: unknown error