made snapshotting much faster, removed unnecessary computing for train/test