Repository navigation

Fix generalization - #100

Draft

SarahAlidoost wants to merge 37 commits into

mainfrom

fix_model

Draft

Fix generalization #100
SarahAlidoost wants to merge 37 commits into
mainfrom
fix_model

Commits on Sep 23, 2026
- check nan of pred in compute_masked_loss in utils
  SarahAlidoost
  committed
  e4bebd7
- change model to predict residuals
  SarahAlidoost
  committed
  06643dc
- improve decoder, add residual_scale to decoder
  SarahAlidoost
  committed
  7f02ef9
Commits on Sep 29, 2026
- make dataloader shuffle false in predict
  SarahAlidoost
  committed
  2c876b4
- fix model copy in train
  SarahAlidoost
  committed
  f2cc65b
Commits on Oct 5, 2026
- refactor train and predict
  SarahAlidoost
  committed
  df34eca
- replace l1_loss with mse to fix generalization
  SarahAlidoost
  committed
  dababfb
- refactor the model
  SarahAlidoost
  committed
  f3185ba
- add decoder_residual_scale to tune
  SarahAlidoost
  committed
  7a871a3
- fix scripts accordingly
  SarahAlidoost
  committed
  0686884
- add optimizer_weight_decay to tune
  SarahAlidoost
  committed
  1a6810b
- fix scripts
  SarahAlidoost
  committed
  10d3ffe
- set calculate_residuals False in data preparation
  SarahAlidoost
  committed
  16ae16e
Commits on Oct 7, 2026
- initialize the conv head in the model
  SarahAlidoost
  committed
  fd03fda
- normalize residuals before loss calculation
  SarahAlidoost
  committed
  1517769
- add res_std to training script
  SarahAlidoost
  committed
  b549d90
- fix training script
  SarahAlidoost
  committed
  0e07312
Commits on Oct 12, 2026
- fix the decoder
  SarahAlidoost
  committed
  94f129b
- fix best_loss in train module
  SarahAlidoost
  committed
  0a61d9f
Commits on Oct 20, 2026
- fix loss averaging in train and predict
  SarahAlidoost
  committed
  c6e4602
- fix saving checkpoint in train
  SarahAlidoost
  committed
  995ff0b
Commits on Oct 27, 2026
Commits on Nov 6, 2026
- fix init file
  SarahAlidoost
  committed
  9874146
- add train_scale to the prediction
  SarahAlidoost
  committed
  f620246
- update tune module
  SarahAlidoost
  committed
  b9cc0e8
- remove unused import
  SarahAlidoost
  committed
  9494e66
- run ruff
  SarahAlidoost
  committed
  8fa6261
Commits on Nov 9, 2026
- fix tune restore code
  SarahAlidoost
  committed
  0e505cb
- fix docstrings
  SarahAlidoost
  committed
  f0f4543
- update code math description
  SarahAlidoost
  committed
  b6c7fbb

Commits

Commits on Sep 23, 2026

Commits on Sep 29, 2026

Commits on Oct 5, 2026

Commits on Oct 7, 2026

Commits on Oct 12, 2026

Commits on Oct 20, 2026

Commits on Oct 27, 2026

Commits on Nov 6, 2026

Commits on Nov 9, 2026