hellozhuo / pidinet

Code for the ICCV 2021 paper "Pixel Difference Networks for Efficient Edge Detection" (Oral).
Other
445 stars 69 forks source link

question about .train() #43

Open ForawardStar opened 1 year ago

ForawardStar commented 1 year ago

I notice that in the function train() of main.py , the model are converted to training mode through model.train(). But, batch normalization or DropOut is not adopted in the network. I wonder is it necessary to use model.train(). Can I abandon it?