Add train, val, test folder paths in data.yaml at save_data_yaml()

Add train, val, test folder paths in data.yaml at save_data_yaml() #902