Skip to content

Commit

Permalink
config: Update config with better defaults
Browse files Browse the repository at this point in the history
  • Loading branch information
braun-steven committed Sep 22, 2023
1 parent fab5043 commit c6ef47b
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions conf/config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ hydra:
# Default set of configurations.
data_dir: "${oc.env:DATA_DIR}/"
results_dir: "${oc.env:RESULTS_DIR}/${project_name}"
project_name: "simple-einet-dc"
project_name: "simple-einet"
lr: 0.0005
batch_size: 64
n_bits: 8
Expand All @@ -33,7 +33,7 @@ max_sigma: 2.0
dry_run: False
seed: ???
log_interval: 10
classification: True
classification: False
device: "cuda"
debug: False
S: 10
Expand All @@ -46,7 +46,7 @@ epochs: 10
load_and_eval: False
cp: False
dist: "normal"
precision: 16
precision: "bf16-mixed"
group_tag: ???
tag: ???
wandb: False
Expand All @@ -55,5 +55,6 @@ profiler: ???
log_weights: False
dataset: ???
num_classes: 10
init_leaf_data: True
init_leaf_data: False
einet_mixture: False
torch_compile: False

0 comments on commit c6ef47b

Please sign in to comment.