Skip to content

Commit 5bf9b15

Browse files
authored
Update README.md
1 parent f07ebd1 commit 5bf9b15

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,8 @@ Useful Tips
112112
Checklist before Running Experiments:
113113
- [ ] Your data should have binary labels **0,1** and **1** is the **minority class** and **0** is the **majority class**
114114
- [ ] Compute the **imbalance_ratio** from your train set and pass it to `AUCMLoss(imratio=xxx)`
115-
- [ ] Choose a proper **initial learning rate** and use `optimizer.update_regularizer(decay_factor=10)` in stagewise
115+
- [ ] Choose a proper **initial learning rate**, e.g., **lr=[0.1, 0.05]** usually works better
116+
- [ ] Use `optimizer.update_regularizer(decay_factor=10)` to update learning rate and regularizer in stagewise
116117
- [ ] Use activation function, e.g., `torch.sigmoid()`, before passing model outputs to loss function
117118
- [ ] Reshape both variables **preds** and **targets** to `(N, 1)` before calling loss function
118119

0 commit comments

Comments
 (0)