- Home /
Question by
kiszkilukiszki · Aug 23, 2021 at 03:46 PM ·
aiunityeditorpython
Stuck at clicking play button with training ml-agents
Hi
I tried to train ml-agents in unity , but when i enter into cmd mlagents-learn trainer_config.yaml --run-id="JumperAI_1" --force
and clicking play button in unity got stuck .
The output in the cmd:
Version information:
ml-agents: 0.27.0,
ml-agents-envs: 0.27.0,
Communicator API: 1.5.0,
PyTorch: 1.9.0+cu102
[INFO] Listening on port 5004. Start training by pressing the Play button in the Unity Editor.
[INFO] Connected to Unity environment with package version 1.9.1-preview and communication version 1.5.0
[INFO] Connected new brain: Jumper?team=0
[WARNING] Deleting TensorBoard data events.out.tfevents.1629732557.DESKTOP-FU8M0IJ.16216.0 that was left over from a previous run.
[WARNING] Deleting TensorBoard data events.out.tfevents.1629732557.DESKTOP-FU8M0IJ.16216.0.meta that was left over from a previous run.
[INFO] Hyperparameters for behavior name Jumper:
trainer_type: ppo
hyperparameters:
batch_size: 128
buffer_size: 2048
learning_rate: 0.0003
beta: 0.005
epsilon: 0.2
lambd: 0.95
num_epoch: 3
learning_rate_schedule: linear
network_settings:
normalize: False
hidden_units: 64
num_layers: 2
vis_encode_type: simple
memory:
sequence_length: 64
memory_size: 128
goal_conditioning_type: hyper
reward_signals:
extrinsic:
gamma: 0.99
strength: 1.0
network_settings:
normalize: False
hidden_units: 128
num_layers: 2
vis_encode_type: simple
memory: None
goal_conditioning_type: hyper
init_path: None
keep_checkpoints: 5
checkpoint_interval: 500000
max_steps: 50000000
time_horizon: 64
summary_freq: 10000
threaded: False
self_play: None
behavioral_cloning: None
C:\Users\Lukasz\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.7_qbz5n2kfra8p0\LocalCache\local-packages\Python37\site-packages\torch\cuda\__init__.py:106: UserWarning:
NVIDIA GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3080 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
I would be very grateful if someone help me.
przechwytywanie.png
(7.9 kB)
Comment