update todo

This commit is contained in:
mayukhdeb
2024-06-20 03:44:42 -07:00
parent 5cc09092a4
commit f64da5d4f1
+1
View File
@@ -26,6 +26,7 @@ todos:
- [x] wandb log loss
- [x] loss go down
[] - Save checkpoint during and after training
[] - inference + visualize examples
"""
import math
import copy