Hi, I have managed to train a model using trainer. I evaluated some results whilst the model was still on the disk using ‘trainer.predict()’. I then used
trainer.save_model() and now want to load it up for usage again. By saving, I got three files in my drive;
I am assuming the model is pytorch_model.bin but I am unsure how do I load it up? I trained a classification model so do I call again;
model = AutoModelForSequenceClassification.from_pretrained(**filepath**, num_labels=5, output_attentions = False, # output_hidden_states = False )
filepath is the path to pytorch_model.bin? do I then call again a trainer and send the model to that to again use trainer.predict() for new predictions?