-
Notifications
You must be signed in to change notification settings - Fork 81
/
Copy pathplot_under_rendering.py
52 lines (42 loc) · 1.7 KB
/
plot_under_rendering.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# Copyright (c) 2024.
# ProrokLab (https://www.proroklab.org/)
# All rights reserved.
"""
Data download example
=====================
This example shows one way of dealing with large data files required for your
examples.
The ``download_data`` function first checks if the data has already been
downloaded, looking in either the data directory saved the configuration file
(by default ``~/.sg_template``) or the default data directory. If the data has
not already been downloaded, it downloads the data from the url and saves the
data directory to the configuration file. This allows you to use the data
again in a different example without downloading it again.
Note that examples in the gallery are ordered according to their filenames, thus
the number after 'plot_' dictates the order the example appears in the gallery.
"""
######################################################################
# Define Hyperparameters
# ----------------------
#
# We set the hyperparameters for our tutorial.
# Depending on the resources
# available, one may choose to execute the policy and the simulator on GPU or on another
# device.
# You can tune some of these values to adjust the computational requirements.
#
import vmas
env = vmas.make_env("dispersion", num_envs=3)
env.get_random_actions()
######################################################################
# Define dwdwd
# ----------------------
#
# We set the hyperparameters for our tutorial.
# Depending on the resources
# available, one may choose to execute the policy and the simulator on GPU or on another
# device.
# You can tune some of these values to adjust the computational requirements.
#
env = vmas.make_env("dispersion", num_envs=3)
env.get_random_actions()