From 3f28bccba36742d3b077a0d1f5b689a68a22cf59 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 4 Feb 2021 01:35:31 -0600
Subject: [PATCH] Fixed the order of pool and relu in pytorch source
---
hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py b/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py
index 5918d960a7..6ddc1b8ea3 100644
--- a/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py
+++ b/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py
@@ -16,10 +16,10 @@ def make_conv_pool_activ(
**conv_kwargs
):
layers = [Conv2d(in_channels, out_channels, kernel_size, **conv_kwargs)]
- if pool_size is not None:
- layers.append(MaxPool2d(pool_size, stride=pool_stride))
if activation:
layers.append(activation())
+ if pool_size is not None:
+ layers.append(MaxPool2d(pool_size, stride=pool_stride))
return layers
--
GitLab