o
    ?߱i                     @   s  d dl mZ d dlmZ d dlmZmZ eedd Z	 eedddid	d
iddiddiddidgeddddedddeddddeddedddeeddeeddd gdddeddd gdddedd ed!dd"edd edd edd edd d#d$ed%d&eed%d%d'd(ed)d(d*d+d,ed-eed%d.d%d/d0gd1d2ed%d.d%d/d0gd1d3d4d5d6id7Z	e
 Ze	fD ]Zd8d9 e  D d  Zejd:d;e ed< qd'S )=    )ConfigStore)LazyDict)MODEL_CHECKPOINTSModelKeyF)post_trainedz/experiment/Stage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted_1_1_rectified_flow_onlyzoverride /modelZ2action_conditioned_video2world_fsdp_rectified_flowzoverride /netZ%cosmos_v1_2B_action_chunk_conditionedzoverride /conditionerZ$action_conditioned_video_conditionerzoverride /data_trainZbridge_13frame_480_640_trainzoverride /data_valZbridge_13frame_480_640_val_self_Z"cosmos_predict2_action_conditionedZcosmos_predict_v2p5Z2b_bridge_action_conditioned)projectgroupnamegh㈵4?g?)lrweight_decayi  zs3://bucket/cosmos_diffusion_v2/official_runs_text2world/Stage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted/checkpoints/iter_000010000/model)enabled)	save_iter	load_pathload_training_statestrict_resumeload_from_object_storesave_to_object_storei     )every_ndo_x0_predictionguidancefpssave_s3)r   d   )	hit_thresr   )every_n_sample_regevery_n_sample_ema
heart_beat
iter_speeddevice_monitorwandb	wandb_10xdataloader_speed)straggler_detection	callbacks   )context_parallel_sizeN      )Z
action_dimZtemporal_compression_ratio)min_num_conditional_framesmax_num_conditional_framesconditional_frames_probsstate_tnet)config         i@  )Zgripper_rescale_factorZnum_action_per_chunkZfps_downsample_ratio
video_size)dataset)
batch_sizesamplerr4   )defaultsjob	optimizer
checkpointtrainermodel_parallelmodeldataloader_trainallow_objectsT)flagsc                 C   s    g | ]\}}|t u r| qS  )_itemlower).0r
   valuerA   rA   Q/data/cameron/vidgen/cosmos-predict2.5/cosmos_predict2/experiments/base/action.py
<listcomp>   s     rG   
experiment_global_)r	   packager
   node)hydra.core.config_storer   +cosmos_predict2._src.imaginaire.lazy_configr   cosmos_predict2.configr   r   DEFAULT_CHECKPOINTdictZ.ac_reason_embeddings_rectified_flow_2b_256_320instancecsrB   globalsitemsexperiment_namestorerA   rA   rA   rF   <module>   s   	&bf
