Skip to content

Commit d012322

Browse files
committed
changed the python function name
Signed-off-by: Bo Wang <[email protected]>
1 parent b1a88d5 commit d012322

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/custom_converters/elu_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def forward(self, x):
1515
return self.elu(x)
1616

1717

18-
def MaxDiff(pytorch_out, trtorch_out):
18+
def cal_max_diff(pytorch_out, trtorch_out):
1919
diff = torch.sub(pytorch_out, trtorch_out)
2020
abs_diff = torch.abs(diff)
2121
max_diff = torch.max(abs_diff)
@@ -43,7 +43,7 @@ def main():
4343
trtorch_out = trt_ts_module(input_data)
4444
print('PyTorch output: \n', pytorch_out[0, :, :, 0])
4545
print('TRTorch output: \n', trtorch_out[0, :, :, 0])
46-
MaxDiff(pytorch_out, trtorch_out)
46+
cal_max_diff(pytorch_out, trtorch_out)
4747

4848

4949
if __name__ == "__main__":

0 commit comments

Comments
 (0)