Remove root-level hip_mace files (now under ckpt/)
Browse files- hip_mace.ckpt +0 -3
- hip_mace.model +0 -3
- hip_mace.yaml +0 -180
hip_mace.ckpt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:3abfeb20ea3e51020e0e95ea0ff0af4d31ee314dc13ffe61ac3f41771adb5d2c
|
| 3 |
-
size 381628939
|
|
|
|
|
|
|
|
|
|
|
|
hip_mace.model
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:ecb968c0ab83bf4d5a3d02ca80a788a36e031cbf53799f0df1d706c3d7b4b898
|
| 3 |
-
size 118399748
|
|
|
|
|
|
|
|
|
|
|
|
hip_mace.yaml
DELETED
|
@@ -1,180 +0,0 @@
|
|
| 1 |
-
E0s: '{1: -13.62222753701504, 6: -1029.4130839658328, 7: -1484.8710358098756, 8: -2041.8396277138045,
|
| 2 |
-
9: -2712.8213146878606}'
|
| 3 |
-
MLP_irreps: 16x0e
|
| 4 |
-
allow_random_padding_pt: true
|
| 5 |
-
amp: null
|
| 6 |
-
amp_dtype: auto
|
| 7 |
-
amsgrad: true
|
| 8 |
-
apply_cutoff: true
|
| 9 |
-
atomic_numbers: '[1, 6, 7, 8, 9]'
|
| 10 |
-
avg_num_neighbors: 12.876944541931152
|
| 11 |
-
batch_size: 64
|
| 12 |
-
beta: 0.9
|
| 13 |
-
beta2: 0.999
|
| 14 |
-
charges_key: REF_charges
|
| 15 |
-
checkpoints_dir: checkpoints/horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 16 |
-
clip_grad: 10.0
|
| 17 |
-
compute_atomic_dipole: false
|
| 18 |
-
compute_avg_num_neighbors: false
|
| 19 |
-
compute_forces: true
|
| 20 |
-
compute_polarizability: false
|
| 21 |
-
compute_stress: false
|
| 22 |
-
config: configs/horm_t1x_hip_pairv2_512_mae.yaml
|
| 23 |
-
config_type_weights: '{"Default":1.0}'
|
| 24 |
-
correlation: 3
|
| 25 |
-
default_dtype: float32
|
| 26 |
-
device: cuda
|
| 27 |
-
dipole_key: dipole
|
| 28 |
-
dipole_weight: 1.0
|
| 29 |
-
distance_transform: None
|
| 30 |
-
distributed: false
|
| 31 |
-
downloads_dir: ./downloads
|
| 32 |
-
dry_run: false
|
| 33 |
-
edge_irreps: null
|
| 34 |
-
elec_temp_key: elec_temp
|
| 35 |
-
ema: false
|
| 36 |
-
ema_decay: 0.999
|
| 37 |
-
embedding_specs: null
|
| 38 |
-
enable_cueq: false
|
| 39 |
-
enable_oeq: false
|
| 40 |
-
energy_key: energy
|
| 41 |
-
energy_weight: 1.0
|
| 42 |
-
eps: 1.0e-12
|
| 43 |
-
error_table: TotalMAEHessian
|
| 44 |
-
eval_interval: 10
|
| 45 |
-
filter_type_pt: none
|
| 46 |
-
final_eval: false
|
| 47 |
-
force_mh_ft_lr: false
|
| 48 |
-
forces_key: forces
|
| 49 |
-
forces_weight: 10.0
|
| 50 |
-
foundation_filter_elements: true
|
| 51 |
-
foundation_head: null
|
| 52 |
-
foundation_model: null
|
| 53 |
-
foundation_model_elements: false
|
| 54 |
-
foundation_model_readout: true
|
| 55 |
-
gate: silu
|
| 56 |
-
head_key: head
|
| 57 |
-
heads: null
|
| 58 |
-
hessian_feature_dim: 512
|
| 59 |
-
hessian_fully_connected: true
|
| 60 |
-
hessian_hidden_irreps: 512x0e + 512x1e + 512x1o + 512x2e
|
| 61 |
-
hessian_key: hessian
|
| 62 |
-
hessian_r_max: 16.0
|
| 63 |
-
hessian_radial_MLP: null
|
| 64 |
-
hessian_weight: 25.0
|
| 65 |
-
hidden_irreps: 512x0e + 512x1e + 512x1o + 512x2e
|
| 66 |
-
hip: true
|
| 67 |
-
hip_scalar_energy_tail: false
|
| 68 |
-
huber_delta: 0.01
|
| 69 |
-
interaction: RealAgnosticResidualInteractionBlock
|
| 70 |
-
interaction_first: RealAgnosticResidualInteractionBlock
|
| 71 |
-
keep_checkpoints: false
|
| 72 |
-
keep_isolated_atoms: false
|
| 73 |
-
launcher: slurm
|
| 74 |
-
lbfgs: false
|
| 75 |
-
les_arguments: null
|
| 76 |
-
log_dir: logs/horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 77 |
-
log_interval: 50
|
| 78 |
-
log_level: INFO
|
| 79 |
-
loss: maeenergyforceshessian
|
| 80 |
-
lr: 0.08
|
| 81 |
-
lr_factor: 0.8
|
| 82 |
-
lr_muon: 0.02
|
| 83 |
-
lr_scheduler_gamma: 0.9993
|
| 84 |
-
max_L: null
|
| 85 |
-
max_ell: 3
|
| 86 |
-
max_num_epochs: 1000
|
| 87 |
-
max_train_samples: null
|
| 88 |
-
max_valid_samples: 1000
|
| 89 |
-
mean: -611.9894409179688
|
| 90 |
-
min_lr: 1.0e-05
|
| 91 |
-
model: MACE
|
| 92 |
-
model_dir: .
|
| 93 |
-
multi_processed_test: false
|
| 94 |
-
multiheads_finetuning: true
|
| 95 |
-
name: horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 96 |
-
num_channels: null
|
| 97 |
-
num_cutoff_basis: 5
|
| 98 |
-
num_interactions: 2
|
| 99 |
-
num_interactions_hessian: 0
|
| 100 |
-
num_radial_basis: 16
|
| 101 |
-
num_samples_pt: 10000
|
| 102 |
-
num_workers: 4
|
| 103 |
-
only_cueq: true
|
| 104 |
-
optimizer: adamw
|
| 105 |
-
override_dirname: mace/cli/run_train.py --config=configs/horm_t1x_hip_pairv2_512_mae.yaml
|
| 106 |
-
--name=horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 107 |
-
--wandb_name=horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 108 |
-
--error_table=TotalMAEHessian --checkpoints_dir=checkpoints/horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 109 |
-
--log_dir=logs/horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 110 |
-
--results_dir=results/horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 111 |
-
--restart_latest
|
| 112 |
-
pair_repulsion: false
|
| 113 |
-
patience: 2048
|
| 114 |
-
pin_memory: true
|
| 115 |
-
plot: true
|
| 116 |
-
plot_frequency: 0
|
| 117 |
-
plot_interaction_e: false
|
| 118 |
-
polarizability_key: polarizability
|
| 119 |
-
polarizability_weight: 1.0
|
| 120 |
-
predict_hessian: true
|
| 121 |
-
pseudolabel_replay: false
|
| 122 |
-
pt_train_file: null
|
| 123 |
-
pt_valid_file: null
|
| 124 |
-
r_max: 7.0
|
| 125 |
-
radial_MLP: '[64, 64, 64]'
|
| 126 |
-
radial_type: bessel
|
| 127 |
-
real_pt_data_ratio_threshold: 0.1
|
| 128 |
-
restart_latest: true
|
| 129 |
-
results_dir: results/horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 130 |
-
samples_per_epoch: 20000
|
| 131 |
-
save_all_checkpoints: true
|
| 132 |
-
save_checkpoints: true
|
| 133 |
-
save_cpu: false
|
| 134 |
-
scaling: rms_forces_scaling
|
| 135 |
-
scheduler: CosineAnnealingLR
|
| 136 |
-
scheduler_patience: 50
|
| 137 |
-
seed: 42
|
| 138 |
-
skip_evaluate_heads: pt_head
|
| 139 |
-
start_swa: null
|
| 140 |
-
statistics_file: null
|
| 141 |
-
std: 0.38076332211494446
|
| 142 |
-
stress_key: REF_stress
|
| 143 |
-
stress_weight: 1.0
|
| 144 |
-
subselect_pt: random
|
| 145 |
-
swa: false
|
| 146 |
-
swa_dipole_weight: 1.0
|
| 147 |
-
swa_energy_weight: 1000.0
|
| 148 |
-
swa_forces_weight: 100.0
|
| 149 |
-
swa_hessian_weight: 1000.0
|
| 150 |
-
swa_lr: 0.001
|
| 151 |
-
swa_polarizability_weight: 1.0
|
| 152 |
-
swa_stress_weight: 10.0
|
| 153 |
-
swa_virials_weight: 10.0
|
| 154 |
-
test_dir: null
|
| 155 |
-
test_file: null
|
| 156 |
-
total_charge_key: total_charge
|
| 157 |
-
total_spin_key: total_spin
|
| 158 |
-
train_file: ./data/ts1x/ts1x_hess_train
|
| 159 |
-
use_agnostic_product: false
|
| 160 |
-
use_embedding_readout: false
|
| 161 |
-
use_last_readout_only: false
|
| 162 |
-
use_layer_norm: false
|
| 163 |
-
use_reduced_cg: false
|
| 164 |
-
use_so3: false
|
| 165 |
-
valid_batch_size: 64
|
| 166 |
-
valid_file: ./data/ts1x/ts1x-val
|
| 167 |
-
valid_fraction: 0.1
|
| 168 |
-
virials_key: REF_virials
|
| 169 |
-
virials_weight: 0.0
|
| 170 |
-
wandb: true
|
| 171 |
-
wandb_dir: null
|
| 172 |
-
wandb_entity: ''
|
| 173 |
-
wandb_log_hypers: null
|
| 174 |
-
wandb_name: horm_t1x_hip_pairv2_512_lr0p08_mae_e1_f10_h25_1000_wu100_cueq_bs64_repro
|
| 175 |
-
wandb_project: hip-mace
|
| 176 |
-
warmup_epochs: 100
|
| 177 |
-
weight_decay: 0.0
|
| 178 |
-
weight_decay_muon: 0.01
|
| 179 |
-
weight_pt_head: 1.0
|
| 180 |
-
work_dir: .
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|