how to use the trained model for other task

[Tranaki]

[hugochan]

@Tranaki sorry for the late response! If you want to use the trained model for inference, you can achieve the goal by commenting out model.train() in the main file. If you want to apply the approach to other tasks, you might need to change the model arch and train the model on that task. Without knowing the task, I am afraid I am not able to provide further help.