Revert "network & training updates"

This reverts commit 36232298951248e6289e066fbf0190b5954d09df.