Tag: GPYTORCH
He visto que la mayoría de los tutoriales/guías tienen el paso de validación fuera del bucle de épocas. Sin embargo, una guía que sigo tiene el paso de validación dentro del bucle de épocas. ¿Cuál es el correcto? Me doy cuenta de que si tienes la validación dentro del bucle . . . Read more
Tengo una pregunta, ¿cómo usar el modelo pre-entrenado guardado? Por ejemplo, tengo un modelo y lo entreno con el conjunto de datos que tengo, luego lo guardo con la extensión .pth. ¿Cómo usar el archivo .pth para probar una nueva imagen? Gracias.
Tengo una tarea de celery que utiliza la biblioteca torch, que internamente utiliza CUDA. Cuando ejecuto la tarea, falla y muestra el siguiente mensaje: “No se puede volver a inicializar CUDA en un subproceso bifurcado. Para usar CUDA con multiprocessing, debes usar el método de inicio ‘spawn’”. Cuando investigué un . . . Read more
¿Alguien puede mejorar el tiempo de procesamiento para generar una lista de números de secuencia? Aquí está mi código y tarda aproximadamente 0.05 segundos. import torch import time import random index = [torch.tensor(660000)] st = time.time() allowed = [x for x in range(index[0])] + [x for x in range(index[0] + . . . Read more
Soy relativamente nuevo en PyTorch y esto podría ser un problema de versión, pero veo que se utiliza torch.load y torch.load_state_dict, y en ambos casos la extensión del archivo es comúnmente “.pth”. Los modelos que he creado puedo guardarlos y cargarlos usando torch.save y torch.load, y luego llamar a model.eval(). . . . Read more