Could you please provide more detailed information about the init_weights in the train.py?
Could you please provide more detailed information about the init_weights in the train.py?