From d5a900caa5f892e97d159390d84fa524e952c313 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Mon, 1 Feb 2021 04:23:18 -0600
Subject: [PATCH] Turned on Softmax layer by default
---
hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py b/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py
index 6f38094d15..5918d960a7 100644
--- a/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py
+++ b/hpvm/test/dnn_benchmarks/pytorch/dnn/_container.py
@@ -25,7 +25,7 @@ def make_conv_pool_activ(
class Classifier(Module):
def __init__(
- self, convs: Sequential, linears: Sequential, use_softmax: bool = False
+ self, convs: Sequential, linears: Sequential, use_softmax: bool = True
):
super().__init__()
self.convs = convs
--
GitLab