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 <#3 (comment)>, 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