Add aesthetic scorer reward function
This commit is contained in:
		| @@ -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) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user