abbeleve/

gender-classification-distilgp2-torch

этот код я делал после предыдущего поскольку в случае если использование торча обязательно для обучения тут нельзя стянуть модель, можно только скачать ее отсюда и запустить

test = torch.load('torch_model_distilbertgpt2.pth', weights_only=False)