13 lines (13 with data), 3.6 kB
B
å®eÀ ã @ s d dl Z dd ZdS )é Nc C s t ¡ } | jdtddd | jdtddd | jdtd dd | jd
tddd | jd
tddd | jdtddd | jdtd dd | jdtddd | jdtddd | jdtddd | jdtddd | jdtdd d | jd!tdd"d | jd#td$d%d | jd&td'd(d | jd)td*d+d | jd,td-d.d | jd/td0d1d | jd2td3d4d | jd5td6d7d | jd8td'd9d | jd:td;d<d | jd=td>d?d | jd@tddAd | jdBtddCd | jdDtdEdFd | jdGtdHdId | jdJtd$dKd | jdLtdMdNd | jdOtddPd | jdQtddRd | jdStddTd | jdUdVtdWdX | jdYdZtd[dX | jd\d]d^d_ | jd`tddad | jdbtdcddd | jdetd;dfd | jdgtdcdhd | jditddjd | jdktdd$dldmgdnd | jdotddpd | jdqtd-drd | jdstddtd | jdutddvd | jdwtdd$dldmgdxd | jdytddzd | jd{tdd$gdzd | jd|td}d}gd~d | jdtddd | jdtddd | jdtddd | jdtdd$d-d
gdd | jdtd dd | jdtddd | d¡}|S )Nz-netÚsamznet type)ÚtypeÚdefaultÚhelpz -baselineÚunetzbaseline net typez-seg_netZ transunetz-modZsam_adptzmod type:seg,cls,val_adz -exp_namezmsa-2dz-typeÚmapz condition type:ave,rand,rand_mapz-visÚ
visualizationz-reverseFzadversary reversez -pretrainz -val_freqéd z interval between each validationz-gpuTzuse gpu or notz-gpu_devicer z
use which gpuz-sim_gpuzsplit sim to this gpuz
-epoch_inié zstart epochz-image_sizei Ú
image_sizez -out_sizeé Úoutput_sizez-patch_sizeé Ú
patch_sizez-dimi Údim_sizez-depthé Údepthz-headsé zheads numberz-mlp_dimÚmlp_dimz-wé z number of workers for dataloaderz-bé zbatch size for dataloaderz-szwhether shuffle the datasetz
-if_warmupzif warm up training phasez-warmup_periodéÈ zwarm up training phasez-lrgü©ñÒMbP?zinitial learning ratez-uinchzinput channel of unetz-imp_lrga2U0*©3?zimplicit learning ratez-weightsz!the weights file you want to testz
-base_weightszthe weights baselinez-sim_weightszthe weights simz-distributedÚnonezmulti GPU ids to use)r r r z-datasetZisiczdataset namez -sam_ckptÚcheckpointszsam checkpoint address)r r z-thdz 3d or notz-chunké` zcrop volume depthz-num_samplezsample pos and negz -roi_sizezresolution of roiz-if_encoder_adapterzif add adapter to encoderz-encoder-adapter-depthsé
é z"the depth of blocks to add adapterz-if_mask_decoder_adapterzif add adapter to mask decoderz-decoder_adapt_depthz the depth of the decoder adapterz-if_encoder_lora_layerzif add lora to encoderz-if_decoder_lora_layerzif add lora to decoderz-encoder_lora_layerzthe depth of blocks to add loraz-if_split_encoder_gpusz!if split encoder to multiple gpusz-devicesz-gpu_fractionsg à?z%how to split encoder to multiple gpusz-if_LST_CNNzif add CNN as side netz-if_LST_encoder_adapterz
if add LSTz-if_LST_decoder_adapterz-encoder_LST_adapter_layersé zwhich layer to add LSTz
-evl_chunkzevaluation chunkz
-data_pathz../datazThe path of segmentation dataÚ ) ÚargparseÚArgumentParserÚadd_argumentÚstrÚintÚboolÚfloatÚlistÚ
parse_args)ÚparserÚopt© r* ú@/Users/hanxuegu/Documents/Documents/bone/bone_paper_codes/cfg.pyr'