Minor fixes to relu blocks test

parent 30d1c887
...@@ -58,7 +58,7 @@ args = parser.parse_args() ...@@ -58,7 +58,7 @@ args = parser.parse_args()
annm_errors = np.zeros(15) annm_errors = np.zeros(15)
appx_errors = np.zeros(15) appx_errors = np.zeros(15)
spatial_relu = torch.nn.ReLU(inplace=True).to(device) spatial_relu = torch.nn.ReLU(inplace=False).to(device)
for f in range(15): for f in range(15):
print('Processsing spatial frequency {}'.format(f)) print('Processsing spatial frequency {}'.format(f))
...@@ -81,8 +81,8 @@ for f in range(15): ...@@ -81,8 +81,8 @@ for f in range(15):
annm_errors[f] += rmse_error(annm_im, true_relu) annm_errors[f] += rmse_error(annm_im, true_relu)
appx_errors[f] += rmse_error(apx_im, true_relu) appx_errors[f] += rmse_error(apx_im, true_relu)
annm_errors /= args.batches * args.batch_size annm_errors /= args.batches
appx_errors /= args.batches * args.batch_size appx_errors /= args.batches
with open(args.output, 'w') as f: with open(args.output, 'w') as f:
f.write('ANNM,APPX\n') f.write('ANNM,APPX\n')
......
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