Сообщество IT-специалистов
Ответы на любые вопросы об IT
Профессиональное развитие в IT
Удаленная работа для IT-специалистов
mat1 and mat2 shapes cannot be multiplied (64x1 and 512x64)
(m x n) и (n x p)
preprocess = transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), ]) torch.manual_seed(42) kwargs = {"nfeat":512, "nhid":64, "nclass":1, "nheads":9, "dropout":0.6, "alpha":0.01} train_dataloader = DataLoader(train_dataset, batch_size=64, shuffle=True) test_dataloader = DataLoader(test_dataset, batch_size=64, shuffle=True)
Сейчас у меня ошибка
Понял, что это проблема в перемножении матриц. Должно быть как-то так
(m x n) и (n x p)
Попробовал изменять вот эти параметры:
Но ошибка та же, меняются только значения в матрицах.
Как правильно это можно исправить?