"name 'F' is not defined" error when training a custom model

[pasan148]

While training a custom model I get this error

NameError                                 Traceback (most recent call last)
<ipython-input-10-6d4c06a1ca45> in <cell line: 2>()
      1 opt = get_config("config_files/en_filtered_config.yaml")
----> 2 train(opt, amp=False)

1 frames
/content/drive/MyDrive/EasyOCR-master/trainer/train.py in validation(model, criterion, evaluation_loader, converter, opt, device)
     71         valid_loss_avg.add(cost)
     72 
---> 73         # calculate accuracy & confidence score
     74         preds_prob = F.softmax(preds, dim=2)
     75         preds_max_prob, _ = preds_prob.max(dim=2)

NameError: name 'F' is not defined

How do I resolve this?

Thank you