Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
38 commits
Select commit Hold shift + click to select a range
56bc779
cleanup + gitignore update
pranavmanivannan Jul 24, 2025
3acd42b
remove unnecessary files
pranavmanivannan Jul 24, 2025
9c06ffb
add modules + remove empty modules + models folder
pranavmanivannan Jul 24, 2025
8d397a9
add GIN and GAT models
pranavmanivannan Jul 24, 2025
111bc87
fix imports and remove unnecessary folder
pranavmanivannan Jul 24, 2025
9182c49
fix dataloader
pranavmanivannan Jul 24, 2025
7fdea22
cleaned prints into logs
pranavmanivannan Jul 24, 2025
7f84cae
added logger utils
pranavmanivannan Jul 24, 2025
8bb4305
change prints into logs
pranavmanivannan Jul 24, 2025
cfb0aab
add training config and related models
pranavmanivannan Jul 24, 2025
252e0ae
update imports
pranavmanivannan Jul 24, 2025
19928bb
add base training script
pranavmanivannan Jul 24, 2025
afa18f7
remove fns that are now in training.py
pranavmanivannan Jul 24, 2025
2ffec6a
update training script
pranavmanivannan Jul 24, 2025
cf61666
keep only models in gat and gin
pranavmanivannan Jul 24, 2025
b140dac
add wandb
pranavmanivannan Jul 24, 2025
68c30bc
remove some todos
pranavmanivannan Jul 24, 2025
54e2883
clean imports
pranavmanivannan Jul 24, 2025
3883acf
clean up more imports
pranavmanivannan Jul 24, 2025
7e27388
add run name arg
pranavmanivannan Jul 25, 2025
d683d31
update log utils filename
pranavmanivannan Aug 5, 2025
66e9fe0
poetry -> uv
pranavmanivannan Aug 5, 2025
4e91c5e
add build system
pranavmanivannan Aug 6, 2025
f059a52
rm unused fn
pranavmanivannan Aug 6, 2025
7def861
rm commented code
pranavmanivannan Aug 6, 2025
b478d84
set single block dwest to -block_id & clean up unused node names var
pranavmanivannan Aug 6, 2025
6b69a39
add saving datasamples as npz
pranavmanivannan Aug 7, 2025
bc7a61f
fix
aaravg04 Aug 10, 2025
5e30c6c
fix fn jax fn + temp config fix
pranavmanivannan Aug 7, 2025
8e6ef09
update config
pranavmanivannan Aug 17, 2025
96ed3c9
use conf model + device
pranavmanivannan Aug 17, 2025
7da1bd6
fix dtype + re-add batch
pranavmanivannan Aug 17, 2025
7959c9c
update jax dep for gpu
pranavmanivannan Aug 17, 2025
c141f7a
enable x64 for higher precision
pranavmanivannan Aug 17, 2025
6bc5297
CFG w/ Ghidra backend
aaravg04 Aug 27, 2025
faf0508
reorganized
aaravg04 Aug 27, 2025
85f99af
mild edits for vp
aaravg04 Aug 30, 2025
3533c33
updated dataprocessing and completed training scripts
aaravg04 Oct 30, 2025
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
15 changes: 13 additions & 2 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,12 @@
out/

# checkpointed model weights
**/CHECK/*
**/*.out

# notebooks
*.ipynb

# Byte-compiled / optimized / DLL files
__pycache__/
*.py[codz]
Expand Down Expand Up @@ -226,5 +235,7 @@ cfg_constructor/out
**/logdat
**/.ipynb_checkpoints
**/*.swp
!**/demo_file_id.txt

**/demo_file_id.txt
**/.vscode
**/weights
**/wandb
70 changes: 0 additions & 70 deletions .ipynb_checkpoints/README-checkpoint.md

This file was deleted.

28 changes: 0 additions & 28 deletions .ipynb_checkpoints/pyproject-checkpoint.toml

This file was deleted.

1 change: 1 addition & 0 deletions .python-version
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
3.11
3 changes: 0 additions & 3 deletions .vscode/settings.json

This file was deleted.

256 changes: 0 additions & 256 deletions GraphIsomorphismNetwork/.ipynb_checkpoints/jax_gin-checkpoint.py

This file was deleted.

Loading