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

Removed criterion assignment to model in config

parent f3cc3c55
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/pasa.py
+ 0
− 2
View file @ 2bc861d4
...@@ -28,5 +28,3 @@ criterion_valid = BCEWithLogitsLoss(pos_weight=empty(1)) ...@@ -28,5 +28,3 @@ criterion_valid = BCEWithLogitsLoss(pos_weight=empty(1))
# model # model
model = PASA(criterion, criterion_valid, optimizer, optimizer_configs) model = PASA(criterion, criterion_valid, optimizer, optimizer_configs)
model.criterion = criterion
model.criterion_valid = criterion_valid
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