Вопросы по теме 'pytorch-ignite'

Optuna Pytorch: возвращаемое значение целевой функции не может быть преобразовано в float
def autotune(trial): cfg= { 'device' : "cuda" if torch.cuda.is_available() else "cpu", # 'train_batch_size' : 64, # 'test_batch_size' : 1000, # 'n_epochs' : 1, # 'seed' : 0,...
537 просмотров
schedule 09.04.2022

DiagonalBiLSTM RuntimeError: логическое значение Tensor с более чем одним значением неоднозначно
Я пытаюсь реализовать Diagonal BiLSTM с помощью фонарика. У меня подготовлена ​​модель CBAM, а также пространственные ворота, как упоминалось в этой статье. Мой код для определения слоя conv2d выглядит следующим образом. def Conv2D(name,...
261 просмотров
schedule 15.10.2022