Some basic refactoring and improvements#5

Open

fentonscode wants to merge 1 commit intomingdeyu:dev1from fentonscode:updates

Commits on Oct 14, 2021

Some quick changes to enable further development and speed up the training loop
fentonscode
committed