Minor fixes to relu blocks test

parent 30d1c887
......@@ -58,7 +58,7 @@ args = parser.parse_args()
annm_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):
print('Processsing spatial frequency {}'.format(f))
......@@ -81,8 +81,8 @@ for f in range(15):
annm_errors[f] += rmse_error(annm_im, true_relu)
appx_errors[f] += rmse_error(apx_im, true_relu)
annm_errors /= args.batches * args.batch_size
appx_errors /= args.batches * args.batch_size
annm_errors /= args.batches
appx_errors /= args.batches
with open(args.output, 'w') as f:
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