Вопросы по теме '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