Uso de clasificador obligatorio en modelo ShiHaotian
JJavier98 opened this issue · 0 comments
JJavier98 commented
Huolas
En el modelo ShiHaotian al inicio del init() se hace una comprobación para que el clasificador o top_module no sea nulo, y si lo es, se asigna un clasificador por defecto.
if self.classifier is None: # <--
self.classifier = ShiHaotian_Classifier(7904, 5, in_features) # <--
De esta manera si quiero añadir un top_module propio, tengo que hardcodear el tamaño de entrada a la salida de la lstm (32) en vez de comprobarlo en tiempo de ejecución.