davidjaw / Domain-adaptation-object-detection-with-YOLOv8

GNU Affero General Public License v3.0
10 stars 2 forks source link

About the train process #2

Closed dusongjie closed 4 months ago

dusongjie commented 5 months ago

I would like to ask, during the training process, when using a custom trainer, after each epoch, is the verification set of the source domain used or the verification of both the target domain and the source domain included?

davidjaw commented 5 months ago

I think it depends on whether you have customize your validate function, i.e., this line. If not, you don't have additional data loader nor domain-separation process, so the validation will directly apply to those data defined in your config.

So the direct answer to your question, after each epoch, is the verification set of the source domain used or the verification of both the target domain and the source domain included?, is NO, only rain domain is evaluated in this case during training.

You can see from here that I just simply validate two domains separately after the training process is complete since I don't wanna modify the validate function lol.