DI-PCG / configs /train /table_train.yaml
thuzhaowang's picture
init
b6a9b6d
raw
history blame contribute delete
325 Bytes
# Train
save_dir: logs/table_train
data_root: /group/40034/wangzhao/data/ipcg/table_new
train_file: train_list_mv_withaug.txt
test_file: test_list_mv.txt
params_dict_file: params_dict.txt
epochs: 200
batch_size: 128
num_workers: 64
lr: 0.0001
seed: 0
logging_iter: 100
ckpt_iter: 10000
# Model
model: DiT_mini
num_params: 19