diff --git a/STATUS.md b/STATUS.md index 6d93e4d..2c878a1 100644 --- a/STATUS.md +++ b/STATUS.md @@ -3,3 +3,5 @@ | iter | val_vol_r2 | delta | action | secs | gpu | change | |------|-----------|-------|--------|------|-----|--------| | 1 | 0.3749 | +0.0928 | KEEP | 2s | gpu=0% vram=10054/12227MiB temp=34°C | iter1 | +| 1 | 0.3011 | +0.0776 | KEEP | 2s | gpu=0% vram=10054/12227MiB temp=34°C | iter1 | +| 2 | 0.3032 | +0.0021 | KEEP | 2s | gpu=0% vram=10054/12227MiB temp=35°C | iter2 | diff --git a/metrics.json b/metrics.json index 51f5c68..94ece46 100644 --- a/metrics.json +++ b/metrics.json @@ -1,10 +1,10 @@ { - "val_vol_r2": 0.22343164905658508, + "val_vol_r2": 0.30321519081159654, "n_test": 275, "knobs": { "WINDOW": 20, - "EMBED_DIM": 32, - "MASK_FRAC": 0.3, + "EMBED_DIM": 64, + "MASK_FRAC": 0.4, "EPOCHS": 200 } } \ No newline at end of file diff --git a/train.py b/train.py index 358e277..efb3bd1 100644 --- a/train.py +++ b/train.py @@ -15,8 +15,8 @@ import torch.nn as nn # --- agent-tunable knobs --- WINDOW = 20 -EMBED_DIM = 32 -MASK_FRAC = 0.30 +EMBED_DIM = 64 +MASK_FRAC = 0.40 EPOCHS = 200 LR = 1e-3 SEED = 0