"torchvision/vscode:/vscode.git/clone" did not exist on "6d774c6fe53f2492e782095bacd544eebfeb0fc5"
Commit 66314a60 authored by Michael Auli's avatar Michael Auli Committed by Myle Ott
Browse files

ATen Fix

parent 173c577b
...@@ -126,5 +126,5 @@ void TemporalConvolutionTBC_backward( ...@@ -126,5 +126,5 @@ void TemporalConvolutionTBC_backward(
} }
auto tmp = dOutput.sum(0, false); auto tmp = dOutput.sum(0, false);
dBias.assign_(tmp.sum(0)); dBias.copy_(tmp.sum(0));
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment