Fixed code for running#2

Khoa-NT wants to merge 4 commits intoNUS-HPC-AI-Lab:mainNUS-HPC-AI-Lab/Neural-Network-Diffusion:mainfrom

Khoa-NT:Fixed-code-for-runningKhoa-NT/Neural-Network-Diffusion:Fixed-code-for-runningCopy head branch name to clipboard