es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Tag: GPYTORCH

DataLoader de pytorch num_workers.

Actualmente estoy viendo este tutorial: https://deeplizard.com/learn/video/kWVgvsejXsE sobre cuál es el valor ideal para num_workers (atributo opcional de la clase DataLoader). Si entiendo bien, si tienes 2 CPUs, una puede utilizarse para cargar los datos y la otra para realizar las tareas (ajustar los pesos, realizar la retropropagación, etc.). Sin embargo, . . . Read more

PyTorch no puede hacer pickle a lambda.

Tengo un modelo que utiliza una LambdaLayer personalizada de la siguiente manera: class LambdaLayer(LightningModule): def __init__(self, fun): super(LambdaLayer, self).__init__() self.fun = fun def forward(self, x): return self.fun(x) class TorchCatEmbedding(LightningModule): def __init__(self, start, end): super(TorchCatEmbedding, self).__init__() self.lb = LambdaLayer(lambda x: x[:, start:end]) self.embedding = torch.nn.Embedding(50, 5) def forward(self, inputs): o = . . . Read more

Cambiar los valores de metadatos en python, Detectron2

Trabajo con Detectron para la segmentación panóptica. Me gustaría cambiar los valores de los metadatos por los valores que he elegido. En particular, me gustaría reemplazar un diccionario por otro que he creado. Sin embargo, obtuve un error: AssertionError: El atributo ‘stuff_dataset_id_to_contiguous_id’ en los metadatos de ‘coco_2017_train_panoptic_separated’ no se puede . . . Read more