Error on train_val file

[adarsh-dikhit]

Issue in train_val.py file

[rndanugerah]

I encountered the same problem. have you solved that problem? @adarsh-dikhit