Ошибка «имя 'F' не определено» при обучении пользовательской модели EasyOCR ⇐ Python

[Anonymous]

Во время обучения пользовательской модели EasyOCR.
Я получаю следующую ошибку:

Код: Выделить всё

NameError                                 Traceback (most recent call last)
 in ()
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

Для обучения модели я использую https://github.com/JaidedAI/EasyOCR/blo ... r/train.py

Код: Выделить всё

import torch.nn.functional as F

Приведенная выше строка кода также включена в скрипт. Но я продолжаю получать эту ошибку.
В чем может быть причина?

Подробнее здесь: https://stackoverflow.com/questions/790 ... stom-model