gabrielkasmi's picture
Update train/train_seg.py
d01dcc3 verified