vitaliykinakh's picture
Upload models, transformations and training configurations
2de08f8 verified
raw
history blame contribute delete
893 Bytes
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