Model Prediction

[BoltonAthitDavies]

How to make a model's prediction on new images? Thanks

[tianyu0207]

How to make a model's prediction on new images? Thanks

you can just change the dataloader and train based on your own data. After that you can simply inference based on your teained model.

[BoltonAthitDavies]

So my problem is I cannot find model variable for using model.load_state_dict. How should I solve it ?

[tianyu0207]

So my problem is I cannot find model variable for using model.load_state_dict. How should I solve it ?

Hi we have just updated the checkpoint on the Github https://github.com/tianyu0207/IGD/blob/main/Encoder_KD_ckpt

[BoltonAthitDavies]

Hi, Sorry for a bunch of questions. For my understanding, Training with p256 will save the model in Encoder_KD_ckpt, Am I right?

Thank tou

ในวันที่ พฤ. 9 มิ.ย. 2022 เวลา 13:58 Yu Tian @.***> เขียนว่า:

So my problem is I cannot find model variable for using model.load_state_dict. How should I solve it ?

Hi we have just updated the checkpoint on the Github https://github.com/tianyu0207/IGD/blob/main/Encoder_KD_ckpt

— Reply to this email directly, view it on GitHub https://github.com/tianyu0207/IGD/issues/3#issuecomment-1150744201, or unsubscribe https://github.com/notifications/unsubscribe-auth/ARJ3SNZNVCVDHZUILAHLIT3VOGIXTANCNFSM5XJ6D7WQ . You are receiving this because you authored the thread.Message ID: @.***>

[xiabc612]

Is there an inference code for new test images? Thanks