diff --git a/src/ptbench/configs/models/alexnet.py b/src/ptbench/configs/models/alexnet.py index ecaee487d1878b1e73bdee8261f06ee475834106..cf8bfd35aa10ad3493be9483ba0347fc8ebb7da1 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 +)