Remove @spaces.GPU decorator, train in background thread directly 6a2c6e2 verified fbzu commited on Apr 26