Final Project for Machine Learning Practical (2024) @ UoE
Find a file
2024-03-04 14:39:58 +00:00
model Updated glue impl, complex loss fn 2024-03-04 04:10:17 +00:00
.gitignore Yay, works in DP via CPU 2024-03-03 21:59:58 +00:00
_ShanghaiA-train.sh Sync 2024-03-03 23:13:57 +00:00
arguments.py Yay, works in DP via CPU 2024-03-03 21:59:58 +00:00
checkpoint.py Added dummy input for one-shot size-checking 2024-02-29 18:44:42 +00:00
dataset.py TEST: train on gt_count instead of kpoint 2024-03-04 01:49:09 +00:00
preprocess_data.py TEST: train on gt_count instead of kpoint 2024-03-04 01:49:09 +00:00
README.md Initial commit 2024-01-30 14:18:26 +00:00
train.py Use mse instead of l1 for info retainment loss, fix mae 2024-03-04 14:39:58 +00:00

mlp-project