ifangcheng

when I set num_classes=1 (255 for background; 0 for person), and train on my own dataset, I always got sub4 =0.000 sub24 =0.000 sub124 =0.000 ans the total loss do...

thanks for reply! i will try to test it, hopefully it works.

I tried the two ways you mentioned, however it both turns out that they have no effects on the gpu assignment. it always used both of the memory of gpu0...