Properly save/load criterion

src/ptbench/models/alexnet.py

src/ptbench/models/densenet.py

src/ptbench/models/logistic_regression.py

src/ptbench/models/pasa.py

src/ptbench/models/signs_to_tb.py

src/ptbench/scripts/predict.py