├── README.md ├── __init__.py ├── dataset.py ├── debug_rollout.ipynb ├── evaluate_gn.py ├── evaluate_gn_rollout.py ├── gen_data.py ├── generate_rollout.py ├── gn_models.py ├── misc ├── actions.png └── test_0.gif ├── myswimmer.py ├── normalizer.py ├── test_normalizer.ipynb ├── train_gn.py ├── utils.py └── visualize.ipynb /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/dataset.py -------------------------------------------------------------------------------- /debug_rollout.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/debug_rollout.ipynb -------------------------------------------------------------------------------- /evaluate_gn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/evaluate_gn.py -------------------------------------------------------------------------------- /evaluate_gn_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/evaluate_gn_rollout.py -------------------------------------------------------------------------------- /gen_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/gen_data.py -------------------------------------------------------------------------------- /generate_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/generate_rollout.py -------------------------------------------------------------------------------- /gn_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/gn_models.py -------------------------------------------------------------------------------- /misc/actions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/misc/actions.png -------------------------------------------------------------------------------- /misc/test_0.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/misc/test_0.gif -------------------------------------------------------------------------------- /myswimmer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/myswimmer.py -------------------------------------------------------------------------------- /normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/normalizer.py -------------------------------------------------------------------------------- /test_normalizer.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/test_normalizer.ipynb -------------------------------------------------------------------------------- /train_gn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/train_gn.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/utils.py -------------------------------------------------------------------------------- /visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fxia22/gn.pytorch/HEAD/visualize.ipynb --------------------------------------------------------------------------------