def reload(self):
config = model.find_config(model.ConfigType.CNN_TRAD_POOL2)
config["n_labels"] = len(self.labels)
self.model = model.SpeechModel(config)
if not self.no_cuda:
self.model.cuda()
self.model.load(self.model_filename)
self.model.eval()
def reload(self):
config = model.find_config(model.ConfigType.EFFB0)
config["n_labels"] = len(self.labels)
self.model = model.SpeechEffModel(config)
if not self.no_cuda:
self.model.cuda()
self.model.load(self.model_filename)
self.model.eval()
print(self.model())
def reload(self):
config = model.find_config(model.ConfigType.RES8_KOOV)
self.model = model.SpeechResModel(config)
self.model.load(self.model_filename)
self.model.eval()