can I train(optimize) on f1 score loss with pytorch

Question

I am building a binary classifier like below. Can I replace the BCELoss to optimize f1 score?

criterion = nn.BCELoss()

preds = model(inputs)
loss = criterion