Is there a prediction code?

[DSQ111]

Is there a prediction code?

[Dootmaan]

Hi @DSQ111 and thank you for your question. Currently there is no dedicated prediction code in this repo, but you can directly borrow the TestModel() function in train.py to build a prediction .py file.