Skip to content
Snippets Groups Projects
Commit 709304f8 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Added missing "pretrained" argument

parent 729a7e9b
No related branches found
No related tags found
1 merge request!4Moved code to lightning
Hide whitespace changes
Inline Side-by-side
src/ptbench/configs/models/alexnet.py
+ 3
− 1
View file @ 709304f8
......@@ -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
)
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