Update

Make code concise

Fix

Delete useless code

Fix bugs

Update code style

Update code

Update code

Fix bug in loading weights

Upload code

Create .gitignore

Initial commit