|
model
|
Updated glue impl, complex loss fn
|
2024-03-04 04:10:17 +00:00 |
|
.gitignore
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
_DDPShA-train.sh
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
_DDPShB-base-train.sh
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
_ShA-base-train.sh
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
_ShA-train.sh
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
_ShB-base-train.sh
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
_ShB-train.sh
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
arguments.py
|
Yay, works in DP via CPU
|
2024-03-03 21:59:58 +00:00 |
|
dataset.py
|
TEST: train on gt_count instead of kpoint
|
2024-03-04 01:49:09 +00:00 |
|
preprocess_data.py
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |
|
README.md
|
Initial commit
|
2024-01-30 14:18:26 +00:00 |
|
train.py
|
Disable img writer
|
2024-03-05 21:27:27 +00:00 |
|
util.py
|
TEST: use tensorboard for stuff
|
2024-03-04 20:32:26 +00:00 |