| data: | |
| path_table: ./data/adult_train.csv | |
| numerical_columns: | |
| - age | |
| - fnlwgt | |
| - education-num | |
| - capital-gain | |
| - capital-loss | |
| - hours-per-week | |
| categorical_columns: | |
| - workclass | |
| - education | |
| - marital-status | |
| - occupation | |
| - relationship | |
| - race | |
| - sex | |
| - native-country | |
| columns_to_drop: null | |
| dropna: true | |
| fillna: false | |
| target_column: label | |
| split_feature_target: true | |
| task: classification | |
| model: | |
| dim: 256 | |
| n_res_blocks: 3 | |
| diffusion: | |
| schedule: quad | |
| n_timesteps: 1000 | |
| target: two_way | |
| trainer: | |
| train_num_steps: 500000 | |
| log_every: 100 | |
| save_every: 10000 | |
| save_num_samples: 64 | |
| max_grad_norm: null | |
| gradient_accumulate_every: 1 | |
| ema_decay: 0.995 | |
| ema_update_every: 10 | |
| lr: 0.0001 | |
| opt_type: adam | |
| opt_params: null | |
| batch_size: 256 | |
| dataloader_workers: 16 | |
| classifier_free_guidance: true | |
| zero_token_probability: 0.1 | |
| fine_tune_from: null | |
| comment: adult_CFG | |