Skip to content

Commit

Permalink
Add pre-trained models
Browse files Browse the repository at this point in the history
  • Loading branch information
cr-xu committed Feb 29, 2024
1 parent 9da3f5c commit 28c1a90
Show file tree
Hide file tree
Showing 64 changed files with 1,001 additions and 1 deletion.
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
cheetah-accelerator>=0.5.18
cheetah-accelerator
gymnasium
imageio==2.4.1
ipywidgets
Expand Down
96 changes: 96 additions & 0 deletions utils/models/Bertha Sparkman/config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
abort_if_off_screen: false
action_mode: direct
batch_size: 100
filter_action:
- 0
- 1
- 3
filter_observation:
- beam
- magnets
frame_stack: null
gamma: 0.99
incoming_mode: random
incoming_values: null
magnet_init_mode: constant
magnet_init_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- &id001 !!python/name:numpy.ndarray ''
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 5
- &id002 !!python/object/apply:numpy.dtype
args:
- f8
- false
- true
state: !!python/tuple
- 3
- <
- null
- null
- null
- -1
- -1
- 0
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==
misalignment_mode: constant
misalignment_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 8
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAA==
n_envs: 40
n_steps: 40
normalize_observation: true
normalize_reward: true
rescale_action: !!python/tuple
- -3
- 3
reward_mode: objective_improvement
run_name: Bertha Sparkman
sb3_device: auto
target_beam_mode: constant
target_beam_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 4
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=
target_mu_x_threshold: .inf
target_mu_y_threshold: .inf
target_sigma_x_threshold: null
target_sigma_y_threshold: null
threshold_hold: 5
time_limit: 25
time_reward: -0.0
total_timesteps: 300000
vec_env: subproc
wandb_run_name: Bertha Sparkman
Binary file added utils/models/Bertha Sparkman/model.zip
Binary file not shown.
Binary file added utils/models/Bertha Sparkman/normalizer
Binary file not shown.
96 changes: 96 additions & 0 deletions utils/models/Betty Gordon/config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
abort_if_off_screen: false
action_mode: delta
batch_size: 100
filter_action:
- 0
- 1
- 3
filter_observation:
- beam
- magnets
frame_stack: null
gamma: 0.99
incoming_mode: random
incoming_values: null
magnet_init_mode: constant
magnet_init_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- &id001 !!python/name:numpy.ndarray ''
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 5
- &id002 !!python/object/apply:numpy.dtype
args:
- f8
- false
- true
state: !!python/tuple
- 3
- <
- null
- null
- null
- -1
- -1
- 0
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==
misalignment_mode: constant
misalignment_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 8
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAA==
n_envs: 40
n_steps: 40
normalize_observation: true
normalize_reward: true
rescale_action: !!python/tuple
- -3
- 3
reward_mode: objective_improvement
run_name: Betty Gordon
sb3_device: auto
target_beam_mode: constant
target_beam_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 4
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=
target_mu_x_threshold: .inf
target_mu_y_threshold: .inf
target_sigma_x_threshold: null
target_sigma_y_threshold: null
threshold_hold: 5
time_limit: 25
time_reward: -0.0
total_timesteps: 300000
vec_env: subproc
wandb_run_name: Betty Gordon
Binary file added utils/models/Betty Gordon/model.zip
Binary file not shown.
Binary file added utils/models/Betty Gordon/normalizer
Binary file not shown.
96 changes: 96 additions & 0 deletions utils/models/David Archibald/config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
abort_if_off_screen: false
action_mode: delta
batch_size: 100
filter_action:
- 0
- 1
- 3
filter_observation:
- beam
- magnets
frame_stack: null
gamma: 0.99
incoming_mode: random
incoming_values: null
magnet_init_mode: constant
magnet_init_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- &id001 !!python/name:numpy.ndarray ''
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 5
- &id002 !!python/object/apply:numpy.dtype
args:
- f8
- false
- true
state: !!python/tuple
- 3
- <
- null
- null
- null
- -1
- -1
- 0
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==
misalignment_mode: constant
misalignment_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 8
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAA==
n_envs: 40
n_steps: 40
normalize_observation: true
normalize_reward: true
rescale_action: !!python/tuple
- -3
- 3
reward_mode: sum_of_pixels
run_name: David Archibald
sb3_device: auto
target_beam_mode: constant
target_beam_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 4
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=
target_mu_x_threshold: .inf
target_mu_y_threshold: .inf
target_sigma_x_threshold: null
target_sigma_y_threshold: null
threshold_hold: 5
time_limit: 25
time_reward: -0.0
total_timesteps: 300000
vec_env: subproc
wandb_run_name: David Archibald
Binary file added utils/models/David Archibald/model.zip
Binary file not shown.
Binary file added utils/models/David Archibald/normalizer
Binary file not shown.
96 changes: 96 additions & 0 deletions utils/models/Gary Buchwald/config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
abort_if_off_screen: true
action_mode: delta
batch_size: 100
filter_action:
- 0
- 1
- 3
filter_observation:
- beam
- magnets
frame_stack: null
gamma: 0.99
incoming_mode: random
incoming_values: null
magnet_init_mode: constant
magnet_init_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- &id001 !!python/name:numpy.ndarray ''
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 5
- &id002 !!python/object/apply:numpy.dtype
args:
- f8
- false
- true
state: !!python/tuple
- 3
- <
- null
- null
- null
- -1
- -1
- 0
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==
misalignment_mode: constant
misalignment_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 8
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
AAAAAAAAAA==
n_envs: 40
n_steps: 40
normalize_observation: true
normalize_reward: true
rescale_action: !!python/tuple
- -3
- 3
reward_mode: objective_improvement
run_name: Gary Buchwald
sb3_device: auto
target_beam_mode: constant
target_beam_values: !!python/object/apply:numpy.core.multiarray._reconstruct
args:
- *id001
- !!python/tuple
- 0
- !!binary |
Yg==
state: !!python/tuple
- 1
- !!python/tuple
- 4
- *id002
- false
- !!binary |
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=
target_mu_x_threshold: .inf
target_mu_y_threshold: .inf
target_sigma_x_threshold: null
target_sigma_y_threshold: null
threshold_hold: 5
time_limit: 25
time_reward: -1.0
total_timesteps: 300000
vec_env: subproc
wandb_run_name: Gary Buchwald
Binary file added utils/models/Gary Buchwald/model.zip
Binary file not shown.
Binary file added utils/models/Gary Buchwald/normalizer
Binary file not shown.
Loading

0 comments on commit 28c1a90

Please sign in to comment.