Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.
Sign upMissing example to save and reload ObjectDetection model #183
Comments
|
That's good question. We should improve the documentation. When calling If you would like to save/load the trained model, please use |
|
Thanks @zhanghang1989 for your response. It seems that save method in detector is not implemented yet? I tried to call the save method after training, but receive an error.
Error message: I take a look at the original detector source of the object detector class and load and save are not implemented yet.
|
|
Thanks for the feedback @rebotnix . We will prioritize the implementation of save/load method. |
I installed the requirements and the demo for object detection works fine.
Is there a way how I can save and reload a trained model after the task.fit call?
detector = task.fit(dataset_train,
num_trials=2,
epochs=epochs,
lr=ag.Categorical(5e-4, 1e-4),
ngpus_per_trial=1,
time_limits=time_limits)
I also tried to set a resume to False to initialize the task to predict directly, but training is always started from the beginng.
?