Refactoring training and inference script to match source code changes#44

vratins merged 3 commits into

maindiff-use/WaterFlow:mainfrom

dev_training_refactordiff-use/WaterFlow:dev_training_refactorCopy head branch name to clipboard