From 709304f8eb4a25da1c9b28ed8682a0ddb3e853cf Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Tue, 11 Apr 2023 13:15:52 +0200
Subject: [PATCH] Added missing "pretrained" argument

---
 src/ptbench/configs/models/alexnet.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ptbench/configs/models/alexnet.py b/src/ptbench/configs/models/alexnet.py
index ecaee487..cf8bfd35 100644
--- a/src/ptbench/configs/models/alexnet.py
+++ b/src/ptbench/configs/models/alexnet.py
@@ -19,4 +19,6 @@ criterion = BCEWithLogitsLoss(pos_weight=empty(1))
 criterion_valid = BCEWithLogitsLoss(pos_weight=empty(1))
 
 # model
-model = Alexnet(criterion, criterion_valid, optimizer, optimizer_configs)
+model = Alexnet(
+    criterion, criterion_valid, optimizer, optimizer_configs, pretrained=False
+)
-- 
GitLab