File size: 1,644 Bytes
8c3ef47 | 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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | {
"accelerator_kwargs": {},
"allow_tf32": true,
"async_reward_computation": false,
"dataset_index": 3,
"exp_name": "o2o",
"global_step": 0,
"high_reward": 1.0,
"huggingface_note": "LOe3000s20_scratch",
"log_with": "wandb",
"logdir": "ll",
"low_reward": 0.0,
"lr_scheduler": "cosine",
"max_iteration": 3000,
"max_workers": 2,
"metric_moving_const": 0.9,
"mixed_precision": "fp16",
"negative_prompts": "",
"num_checkpoint_limit": 10,
"num_epochs": 6,
"offpolicy_sample_batch_size": 1,
"online_multification_number": 1,
"per_prompt_stat_tracking": false,
"per_prompt_stat_tracking_buffer_size": 16,
"per_prompt_stat_tracking_min_count": 16,
"project_kwargs": {
"automatic_checkpoint_naming": true,
"logging_dir": "./outputs/logs",
"project_dir": "./outputs/",
"total_limit": 5
},
"resolution": 768,
"resume_from": "",
"run_name": "",
"sample_eta": 1.0,
"sample_guidance_scale": 5.0,
"sample_num_batches_per_iteration": 1,
"sample_num_steps": 16,
"save_freq": 400,
"save_start": 100,
"seed": 1,
"show_metrics": true,
"test_mode": true,
"tracker_kwargs": {},
"tracker_project_name": "O2",
"train_adam_beta1": 0.9,
"train_adam_beta2": 0.999,
"train_adam_epsilon": 1e-08,
"train_adam_weight_decay": 0.0001,
"train_adv_clip_max": 5.0,
"train_batch_size": 2,
"train_cfg": true,
"train_clip_range": 0.0001,
"train_gradient_accumulation_steps": 1,
"train_learning_rate": 0.0005,
"train_max_grad_norm": 1.0,
"train_num_inner_iterations": 1,
"train_timestep_fraction": 1.0,
"train_use_8bit_adam": false,
"usage_dataset_size": 500
} |