Setup Training Environment

Setup Training Environment