Hola!! El código me funciona perfecto, pero en la parte del entrenamiento tengo el error de que la pérdida es "nan", al revisar los errores y printear varias cosas descubrí que el error esta en el rango de los datos o algo similar. Puede ser por la manera en que decidí separar las variables como numéricas y categóricas, pero en ellas no tengo nada que sea nan.
El problema surge en el entrenamiento, pero viene desde la defincion de la funcion forward del TabularMLP en su primera linea:
def forward(self, x_num, x_cat):
x = [embedding(x_cat[:, i]) for i, embedding in enumerate(self.embeddings)]
error con gpu: RuntimeError: CUDA error: device-side assert triggered
error con cpu: IndexError: index out of range in self
Hola!! El código me funciona perfecto, pero en la parte del entrenamiento tengo el error de que la pérdida es "nan", al revisar los errores y printear varias cosas descubrí que el error esta en el rango de los datos o algo similar. Puede ser por la manera en que decidí separar las variables como numéricas y categóricas, pero en ellas no tengo nada que sea nan. El problema surge en el entrenamiento, pero viene desde la defincion de la funcion forward del TabularMLP en su primera linea:
def forward(self, x_num, x_cat): x = [embedding(x_cat[:, i]) for i, embedding in enumerate(self.embeddings)]
error con gpu: RuntimeError: CUDA error: device-side assert triggered error con cpu: IndexError: index out of range in self