a je7 @sUddlZddlmZddlmZmZmZmZeGdddZiZ ee efe d<iZ ee e fe d<de d <ee d <d e d <ed d d dddde d <de d<ed ddddddddd e d<de d<ed ddddd d ddddd e d<ej e e ZdS)!N) dataclass)TupleLiteralDictOptionalc@seZdZUdZeed<dZeeed<dZee ed<dZ e ed<d Z eeed <d Z ee ed <d Z eed<dZeed<dZeded<dZeed<dZeed<dZeed<dZeed<dZeed<dZeed<dZeed<d Zeed!<d"Zeeed#<dZeed$<d%Zeed&<d'Zeed(<d)Z eed*<d)Z!eed+<d,Z"eed-<d.Z#eed/<dZ$eed0<dZ%eed1<d"Z&eeed2<d3Z'e ed4<d3Z(e ed5<d"S)6Options input_size)@rr down_channels)FFFTTTdown_attentionT mid_attention)r r r r up_channels)TTTF up_attentionr splat_size output_sizes3 data_modeǧH@fovyg?znearg@zfar num_viewsnum_input_viewsg? cam_radius num_workersz ./workspace workspaceNresume batch_sizegradient_accumulation_steps num_epochsg? lambda_lpips gradient_clipbf16mixed_precisiong-C6:?lrprob_grid_distortionprob_cam_jitter test_pathFforce_cuda_rast fancy_video))__name__ __module__ __qualname__r int__annotations__rrrboolrrrrrrrrfloatrrrrrr r!strr"rr#r%r'r(r)r+r,r-r.r/r0r1r:r:/data/tang/glrm/core/options.pyrs<                       rconfig_defaults config_doczthe default settings for LGMlrmz+small model with lower resolution Gaussianssmallrr rr$r*)r rrr#r%r+z*big model with higher resolution Gaussiansbig)r r r rr )TTTFFr r ) r rrrrr#rr%r+ztiny model for ablationtiny) r r rr )FFFFT)r rr )TFFF) r rrrrrrr#rr%r+)tyro dataclassesrtypingrrrrrr<r9r6r=extrassubcommand_type_from_defaults AllConfigsr:r:r:r;sX  E