diff --git a/src/ptbench/configs/models/pasa.py b/src/ptbench/configs/models/pasa.py index c02d650dcaa03bd4c9aef47ca2d34b858ecd3e73..3ee0b92164b5531b65049b94e71b01b07e2ad27e 100644 --- a/src/ptbench/configs/models/pasa.py +++ b/src/ptbench/configs/models/pasa.py @@ -28,5 +28,3 @@ criterion_valid = BCEWithLogitsLoss(pos_weight=empty(1)) # model model = PASA(criterion, criterion_valid, optimizer, optimizer_configs) -model.criterion = criterion -model.criterion_valid = criterion_valid