pytorch

Вопрос или проблема Мне нужно реализовать кастомный DataLoader, который наследуется от torch.data.utils.DataLoader. Я искал полчаса, но не нашел ни примеров, ни документации об этом. Какие методы мне нужно реализовать? И какие типы данных мне следует

Вопрос или проблема Контекст Я пытаюсь применить обучение с подкреплением к трансформеру. У меня есть следующие токены: ["<sos>", "<eos>", "roses", "are", "red"]. Моя цель – использовать основанную на трансформере политику сети для генерации токенов ["<

Вопрос или проблема Я пытаюсь провести инференс модели обнаружения объектов. У меня есть несколько камер с одинаковым сценарием использования, поэтому я использую многопоточность для обработки этого. У меня есть объект Model().

Вопрос или проблема Я пытаюсь реализовать модель wide and deep от Google, и у меня есть вопрос о ее выходе. Согласно уравнению (3) в статье: $$ P(Y=1|X) = \sigma(w_{wide}^T[x,\phi(x)] + w_{deep}^T a^{(l_f)+b} $$ Должен ли $P(Y=1|X)$ быть скалярным?

Вопрос или проблема Я пытаюсь реализовать нейронную сеть с остаточным распространением в Rust, используя tch-rs (PyTorch). Пока что вот мой код: fn res_block(vs: &nn::Path) -> impl ModuleT { let mut default = ConvConfigND::default();

Вопрос или проблема Я пытаюсь запустить обучение модели, над которой работаю, но по какой-то причине Google Collab не хочет использовать более 1% GPU. Я нашел похожую тему, где говорилось, что мне следует понизить версию CUDA с 12.

Вопрос или проблема Код, который я написал: import faiss import os import numpy as np import torch from torchvision.models import resnet50 from torchvision.models import ResNet50_Weights from torchvision import transforms from PIL import Image accepted_formats = [".