-
Notifications
You must be signed in to change notification settings - Fork 6
/
saving_loading_device_config.py
30 lines (23 loc) · 1.06 KB
/
saving_loading_device_config.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
"""Examples of how to save and load an environment's device configuration from file.
To make experiments repeatable and compare algorithms, you may wish to fix UE positions.
This can be achieved by saving device configurations to an editable JSON file and
reloading the same configuration each experiment.
GymD2D uses Python's inbuilt pathlib paths to refer to files as it simplifies file handling.
"""
from pathlib import Path
import gym
import gym_d2d
# SAVING DEVICE CONFIG
# --------------------
# create a new environment and use `reset()` to generate random UE positions
env = gym.make('D2DEnv-v0')
env.reset()
# save the existing device configuration to a JSON file
env.save_device_config(Path.cwd() / 'device_config.json')
# LOADING DEVICE CONFIG
# ---------------------
# initialise a new environment with a Pathlib path to a device config file
env_config = {'device_config_file': Path.cwd() / 'device_config.json'}
env = gym.make('D2DEnv-v0', env_config=env_config)
# calls to `reset()` will use the device configuration from file---if the device ID is in it.
env.reset()