Can not load model

[mijosch]

for saving:

class MyModelSave(Callback): def on_train_epoch_end(self, trainer, pl_module): torch.save(pl_module.state_dict(),"model.mdl")

for loading: checkpoint = torch.load('model.mdl') print("loaded") tft.load_state_dict(checkpoint) print("tft from statedict")

Model never get loaded(waited for 4 hours), statedict size > 250mb

How do you save and load the TemporalFusionTransformer? This does not work.

[adamwuyu]

Hi @mijosch I met the same problem, have you figured how to save and load TFT modle?

[nicocheh]

same here, do you have any news on this?

[ozanozyegen]

I use load_from_checkpoint function as shown in the TFT tutorial. best_tft = tft.load_from_checkpoint(best_model_path) There are alternative ways to do it described in the lightning docs.