Thank you! This worked for me.

@alankbi Hi, I have a question when I run this code : { start_time = time.time() m.trainer.fit(m) print(f"— Training on GPU: {(time.time() - start_time):.2f} seconds —") print(m.trainer.fit(m))

import tempfile tmpdir = tempfile.TemporaryDirectory()

#save the prediction dataframe after training and compare with prediction after reload checkpoint img_path = get_data(“/content/orthomosaic.jpg”) m.create_trainer() pred_after_train = m.predict_image(path = img_path)

#Create a trainer to make a checkpoint m.trainer.save_checkpoint(“/content/checkpoint.pl”)

m.trainer.save_checkpoint(“/content/checkpoint.pl”,format(tmpdir))

#reload the checkpoint to model object after = main.deepforest.load_from_checkpoint(“{}/checkpoint.pl”.format(tmpdir)) pred_after_reload = after.predict_image(path = img_path)

assert not pred_after_train.empty assert not pred_after_reload.empty pd.testing.assert_frame_equal(pred_after_train,pred_after_reload) } I get this error: AttributeError: ‘NoneType’ object has no attribute ‘state_dict’ would you please help me?