Add aesthetic scorer reward function

This commit is contained in:
Kevin Black
2023-06-27 10:40:36 -07:00
parent 8cab96dea4
commit bae3f43f5f
6 changed files with 68 additions and 2 deletions

View File

@@ -42,7 +42,7 @@ def main(_):
)
if accelerator.is_main_process:
accelerator.init_trackers(project_name="ddpo-pytorch", config=config.to_dict())
logger.info(config)
logger.info(f"\n{config}")
# set seed
set_seed(config.seed, device_specific=True)