Skip to content

Commit

Permalink
adding resume to the inits on learners
Browse files Browse the repository at this point in the history
  • Loading branch information
Kaiyotech committed Jan 25, 2023
1 parent 842a217 commit a2fa46b
Show file tree
Hide file tree
Showing 9 changed files with 10 additions and 1 deletion.
1 change: 1 addition & 0 deletions learner_aerial.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"], db=Constants_aerial.DB_NUM) # host="192.168.0.201",
redis.delete("worker-ids")
Expand Down
1 change: 1 addition & 0 deletions learner_ceil_pinch.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"], db=Constants_ceil_pinch.DB_NUM) # host="192.168.0.201",
redis.delete("worker-ids")
Expand Down
1 change: 1 addition & 0 deletions learner_flick.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"],
db=Constants_flick.DB_NUM) # host="192.168.0.201",
Expand Down
1 change: 1 addition & 0 deletions learner_flip_reset.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"],
db=Constants_flip_reset.DB_NUM) # host="192.168.0.201",
Expand Down
3 changes: 2 additions & 1 deletion learner_gp.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"],
db=Constants_gp.DB_NUM) # host="192.168.0.201",
Expand Down Expand Up @@ -147,7 +148,7 @@
disable_gradient_logging=True,
)

alg.load("GP_saves/Opti_1674312981.1218257/Opti_16840/checkpoint.pt")
alg.load("GP_saves/Opti_1674573038.6569872/Opti_17220/checkpoint.pt")
alg.agent.optimizer.param_groups[0]["lr"] = logger.config.actor_lr
alg.agent.optimizer.param_groups[1]["lr"] = logger.config.critic_lr

Expand Down
1 change: 1 addition & 0 deletions learner_kickoff.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"], db=0) # host="192.168.0.201",
redis.delete("worker-ids")
Expand Down
1 change: 1 addition & 0 deletions learner_pinch.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"], db=Constants_pinch.DB_NUM) # host="192.168.0.201",
redis.delete("worker-ids")
Expand Down
1 change: 1 addition & 0 deletions learner_recovery.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"],
db=Constants_recovery.DB_NUM) # host="192.168.0.201",
Expand Down
1 change: 1 addition & 0 deletions learner_selector.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@
id=run_id,
config=config,
settings=wandb.Settings(_disable_stats=True, _disable_meta=True),
resume=True,
)
redis = Redis(username="user1", password=os.environ["redis_user1_key"],
db=Constants_selector.DB_NUM) # host="192.168.0.201",
Expand Down

0 comments on commit a2fa46b

Please sign in to comment.