Update README.md

update Jupiter Notebook

fix ValueError raising

remove redundant code

remove redundant code

fix tensor2numpy is not definded

add tensor2numpy and to_cpu

add main function and default ckpt-path

add test.py

add main function and remove redundant code

add main function and remove redundant code

update test set

Update README.md

Created using Colaboratory

Update README.md