Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Energy forces #278

Merged
merged 37 commits into from
Sep 20, 2024
Merged
Show file tree
Hide file tree
Changes from 34 commits
Commits
Show all changes
37 commits
Select commit Hold shift + click to select a range
7b56afd
Energy and Force Prediction changes (loss function in base, and optio…
RylieWeaver Sep 6, 2024
a277ee5
comments and renamings
RylieWeaver Sep 6, 2024
fd248b5
Black formatting and mak computational graph fixes
RylieWeaver Sep 6, 2024
c3eaedc
fix loss weighting
RylieWeaver Sep 6, 2024
5136b89
black formatting
RylieWeaver Sep 6, 2024
65f6f1d
black formatting
RylieWeaver Sep 6, 2024
3395484
black formatting
RylieWeaver Sep 6, 2024
c5d0c7b
Fix DIMEStack testing issues, and compute_grad_energy default in conf…
RylieWeaver Sep 6, 2024
fc5a554
LJ example added first draft
RylieWeaver Sep 16, 2024
0bce3d5
formatting
RylieWeaver Sep 16, 2024
271e36a
formatting
RylieWeaver Sep 16, 2024
9de9677
formatting
RylieWeaver Sep 16, 2024
a904def
formatting
RylieWeaver Sep 16, 2024
12c042e
formatting
RylieWeaver Sep 16, 2024
0f3fe54
formatting
RylieWeaver Sep 16, 2024
4df57c7
take out images and adjust unecessary import
RylieWeaver Sep 16, 2024
4779309
revert to SiLU in DimeNet
RylieWeaver Sep 16, 2024
456ebf7
don't create plots by default
RylieWeaver Sep 16, 2024
20f2e41
file cleanup and dataset test
RylieWeaver Sep 17, 2024
2da742e
unnecessary import
RylieWeaver Sep 17, 2024
70206dc
file cleanup
RylieWeaver Sep 17, 2024
7032acd
add some tests back in
RylieWeaver Sep 17, 2024
f98fb3e
Restore all tests
RylieWeaver Sep 17, 2024
d66de16
formatting
RylieWeaver Sep 17, 2024
206b52f
check dataset things
RylieWeaver Sep 17, 2024
9980834
formatting
RylieWeaver Sep 17, 2024
fa10c20
Revise paths to be more succinct and Use radius from config
RylieWeaver Sep 17, 2024
42551b8
file restructuring and using hydra radius graph function
RylieWeaver Sep 17, 2024
ee20cf7
formatting
RylieWeaver Sep 17, 2024
1fa7766
renaming
RylieWeaver Sep 17, 2024
e8a4a4d
remove qm9 test
RylieWeaver Sep 17, 2024
5d358a3
remove qm9 test
RylieWeaver Sep 17, 2024
06dad51
smaller number samples for test
RylieWeaver Sep 17, 2024
393976c
Update examples/LennardJones/LJ_inference_plots.py
allaffa Sep 17, 2024
85f44ab
use info function
RylieWeaver Sep 17, 2024
81992af
Unnecessary __init__ file
RylieWeaver Sep 17, 2024
bc93b9b
Unecessary json args
RylieWeaver Sep 17, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
75 changes: 75 additions & 0 deletions examples/LennardJones/LJ.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,75 @@
{
"Verbosity": {
"level": 2
},
"Dataset": {
"name": "LJdataset",
"path": {"total": "./dataset/data"},
"format": "XYZ",
"rotational_invariance": true,
"node_features": {
"name": ["atom_type"],
"dim": [1],
"column_index": [0]
},
"graph_features":{
"name": ["total_energy"],
"dim": [1],
"column_index": [0]
}
},
"NeuralNetwork": {
"Architecture": {
"periodic_boundary_conditions": true,
"model_type": "DimeNet",
"equivariance": false,
"radius": 5.0,
"max_neighbours": 5,
"int_emb_size": 32,
"out_emb_size": 16,
"basis_emb_size": 8,
"num_before_skip": 1,
"num_after_skip": 1,
"envelope_exponent": 5,
"num_radial": 5,
"num_spherical": 2,
"hidden_dim": 20,
"num_conv_layers": 4,
"output_heads": {
"node": {
"num_headlayers": 2,
"dim_headlayers": [60,20],
"type": "mlp"
}
},
"task_weights": [1]
},
"Variables_of_interest": {
"input_node_features": [0],
"output_index": [
0
],
"type": [
"node"
],
"output_dim": [1],
"output_names": ["graph_energy"]
},
"Training": {
"num_epoch": 15,
"batch_size": 64,
"patience": 20,
"early_stopping": true,
"Optimizer": {
"type": "Adam",
"learning_rate": 0.005
},
"conv_checkpointing": false
}
},
"Visualization": {
"plot_init_solution": true,
"plot_hist_solution": true,
"create_plots": true
}
}
Loading
Loading