__pycache__ pose_env_1 testing vit_env vit_test test_vit_model.ipynb models models_2