data: | |
path_table: ./data/housing_train.csv | |
numerical_columns: | |
- longitude | |
- latitude | |
- housing_median_age | |
- total_rooms | |
- total_bedrooms | |
- population | |
- households | |
- median_income | |
- median_house_value | |
categorical_columns: | |
- ocean_proximity | |
columns_to_drop: null | |
dropna: true | |
fillna: false | |
target_column: median_house_value | |
split_feature_target: true | |
task: regression | |
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: housing_CFG | |