Помимо этого, вы можете представить нам немного меньший пример и немного подробнее определить вашу проблему. Если что-то не так, оставьте комментарий с подробным объяснением проблемы (например, почему вы полагаете, что ваша реализация неверна).

код пытается использовать пользовательскую реализацию dropout: %reset -f import torch import torch.nn as nn # import torchvision # import torchvision.transforms as transforms import torch import torch.nn as nn import torch.utils.data as ...