Files
diffusionNAG/NAS-Bench-201/configs/transfer_nag_config.pt

26 lines
3.4 KiB
Plaintext
Raw Normal View History

2024-03-20 11:21:34 +00:00
PKve_meta_nas/data.pklFB
ZZZZZZZZZZ<5A>cml_collections.config_dict.config_dict
ConfigDict
q)<29>q}q(X_fieldsq}q(X search_spaceqNXresumeq<06>X folder_nameqX0404qXtaskq Xtr_meta_predictorq
Xexp_nameq NX
model_typeq Xmeta_predictorq Xscorenet_ckpt_pathqX*checkpoints/score_model/model_best.pth.tarqXis_metaq<10>Xtrainingqh)<29>q}q(h}q(XsdeqXvesdeqX
continuousq<17>X reduce_meanq<18>Xnoisedq<19>X
batch_sizeqMXeval_batch_sizeqKdXn_itersqM'X snapshot_freqqM<>Xlog_freqqKdX eval_freqqKdXsnapshot_freq_for_preemptionq M<>Xsnapshot_samplingq!<21>Xlikelihood_weightingq"<22>Xt_spotq#G?<3F>Xload_pretrainedq$<24>Xpretrained_model_pathq%huX_lockedq&<26>X
_type_safeq'<27>X _convert_dictq(<28>ubXsamplingq)h)<29>q*}q+(h}q,(Xmethodq-Xpcq.X predictorq/Xeuler_maruyamaq0X correctorq1Xlangevinq2Xrtolq3G><3E><><EFBFBD><EFBFBD><EFBFBD>h<EFBFBD>Xatolq4G><3E><><EFBFBD><EFBFBD><EFBFBD>h<EFBFBD>X
ode_methodq5Xdopri5q6Xode_stepq7G?<3F>z<EFBFBD>G<EFBFBD>{X n_steps_eachq8KX noise_removalq9<71>Xprobability_flowq:<3A>Xsnrq;G?<3F>z<EFBFBD>G<EFBFBD>{Xvis_rowq<KXvis_colq=KXclassifier_scaleq>G?<3F>Xregressq?<3F>Xlabelsq@XmaxqAX weight_ratioqB<71>Xweight_schedulingqC<71>h#G?<3F>X
t_spot_endqDGXnumber_chain_stepsqEK2Xcheck_datanameqFXcifar10qGuh&<26>h'<27>h(<28>ubXevalqHh)<29>qI}qJ(h}qK(X
begin_ckptqLKXend_ckptqMKhMXenable_samplingqN<71>X num_samplesqOMX mmd_distanceqPXRBFqQX max_subgraphqR<71>X
save_graphqS<71>uh&<26>h'<27>h(<28>ubXdataqTh)<29>qU}qV(h}qW(XcenteredqX<71>XdequantizationqY<71>XrootqZX#./data/processed/NASBench201_val.ptq[Xnameq\X NASBench201q]X split_ratioq^G?<3F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>X dataset_idxq_Xrandomq`Xmax_nodeqaKXn_vocabqbKX
START_TYPEqcKXEND_TYPEqdKX
num_graphsqeM =X num_channelsqfKX except_inoutqg<71>Xtriu_adjqh<71>X connect_prevqi<71>X
label_listqj]qkXmeta-accqlaX aug_mask_algoqmXfloydqnX num_trainqoK<6F>uh&<26>h'<27>h(<28>ubXmodelqph)<29>qq}qr(h}qs(h\XMetaNeuralPredictorqtXema_ratequG?<3F><>.H<><48>X normalizationqvX GroupNormqwX nonlinearityqxXswishqyXnfqzK<7A>Xnum_gnn_layersq{KX size_condq|<7C>Xembedding_typeq}X
positionalq~Xrw_depthqKX graph_layerq<72>X PosTransLayerq<72>Xedge_thq<68>G<EFBFBD><47>Xheadsq<73>KX
attn_clampq<70><71>X
input_typeq<65>XDAq<41>Xhsq<73>MXnzq<7A>K8X
num_sampleq<65>KX
num_scalesq<73>M<EFBFBD>Xbeta_minq<6E>G?<3F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Xbeta_maxq<78>G@4X sigma_minq<6E>G?<3F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>X sigma_maxq<78>G@4Xdropoutq<74>G?<3F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>X graph_encoderq<72>h)<29>q<EFBFBD>}q<>(h}q<>(Xinitial_hiddenq<6E>KX
gcn_hiddenq<6E>K<EFBFBD>X
gcn_layersq<73>KX linear_hiddenq<6E>K<EFBFBD>uh&<26>h'<27>h(<28>ubuh&<26>h'<27>h(<28>ubXoptimq<6D>h)<29>q<EFBFBD>}q<>(h}q<>(X weight_decayq<79>KX optimizerq<72>XAdamq<6D>Xlrq<72>G><3E><><EFBFBD><EFBFBD><EFBFBD>h<EFBFBD>Xbeta1q<31>G?<3F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Xepsq<73>G>Ey<45><79>0<EFBFBD>:Xwarmupq<70>M<EFBFBD>X grad_clipq<70>G?<3F>uh&<26>h'<27>h(<28>ubXseedq<64>K*Xdeviceq<65>ctorch
device
q<EFBFBD>Xcudaq<61>K<00>q<EFBFBD>Rq<52>Xlogq<67>h)<29>q<EFBFBD>}q<>(h}q<>(X use_wandbq<62><71>Xwandb_project_nameq<65>XGTAD_v1q<31>Xlog_valid_sample_propq<70><71>Xnum_graphs_to_visualizeq<65>Kuh&<26>h'<27>h(<28>ubuh&<26>h'<27>h(<28>ub.PK<08><>R~  PK#ve_meta_nas/versionFBZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ3
PKўgUPK<00><>R~  ve_meta_nas/data.pklPKўgUl ve_meta_nas/versionPK,-<00><00> PKU PK<00><00>