Project structure of: openai/Video-Pre-Training
agent.pyConfigure Minecraft agent for reinforcement learningbehavioural_cloning.pyTrains actor-critic model for behavioral cloning.data_loader.pyData loader for batch processing with worker threadsinverse_dynamics_model.pyMinecraft action predictor with IDM model.action_head.pyActionHead: Reinforcement learning action heads abstract base classaction_mapping.pyAction mapping for video game inputs and management.actions.pyMinecraft actions, quantization, and transformer.impala_cnn.pyImpalaCNN: Customizable, Multi-Stack Classification Modelmasked_attention.pyMasked Attention for time series data.minecraft_util.pyCompute entropy from categorical head outputs.misc.pyData processing tasks and function design discussed.mlp.pyMLP class: Neural network with specified layers, normed linear.normalize_ewma.pyNormalizes data, maintains running mean & variance.policy.pyMinecraftAgentPolicy with reinforcement learning, image preprocessingscaled_mse_head.pyScaled MSE head for linear output layer and MSE loss.torch_util.pyTorch utilities: libraries, device defaults, tensor functions.tree_util.pyPyTree API utility functions and data type registration.util.pyNeural network utilities for data processingxf.pyAttention mechanism, layers for transformer models.
README.mdMinecraft AI training, competition, resource data, 10-min house building.requirements.txtInstalls essential libraries for Python.run_agent.pyRun MineRL model with pre-trained weights.run_inverse_dynamics_model.pyInverse dynamics model with game inputs and video streaming.