YOLOX icon indicating copy to clipboard operation
YOLOX copied to clipboard

Can i freeze the class based loss so that i can train only class what i want?

Open rahulsharma11 opened this issue 3 years ago • 4 comments

Hi, I trained the YOLOX_m with 7 classes and it worked well. Now i want to train only 2 classes (out of those 7 classes) with some more dataset but keeping other 5 classes untouched.

Please suggest if i can do that and if yes then how?

Thanks.

rahulsharma11 avatar Dec 29 '21 04:12 rahulsharma11

You could masked your loss to train those classes.

FateScript avatar Dec 29 '21 09:12 FateScript

Hi, And how can i do that...can you suggest few steps? Thanks.

rahulsharma11 avatar Dec 29 '21 09:12 rahulsharma11

Hi, FateScript. Can you do me a favor in adding me on the official wechat group?

frandoLin avatar Dec 29 '21 10:12 frandoLin

Hi, And how can i do that...can you suggest few steps? Thanks. Hello, Did you finally make it? I have the same problem as you, could you give me some advice?

Finorita avatar Jun 27 '22 02:06 Finorita