Define export formats model.names attribute (#9474)

This commit is contained in:
Glenn Jocher 2024-04-01 18:53:44 +02:00 committed by GitHub
parent 80c59e2461
commit 2cee8893d9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -735,7 +735,10 @@ class Model(nn.Module):
if hasattr(self.model, "names"):
return check_class_names(self.model.names)
elif self.predictor:
else:
if not self.predictor: # export formats will not have predictor defined until predict() is called
self.predictor = self._smart_load("predictor")(overrides=self.overrides, _callbacks=self.callbacks)
self.predictor.setup_model(model=self.model, verbose=False)
return self.predictor.model.names
@property