alicezhai's picture
Update modules/train_func.py
0ab1e5f verified