Commit Graph

10 Commits (a87e90e0dd8a22c530a52ccdbf2eb440b02ae212)

Author SHA1 Message Date
aria1th a87e90e0dd clean up code partially 2022-11-27 12:36:21 +09:00
aria1th ad395f11b8 fix DivisionByZero and save 2022-11-27 11:07:38 +09:00
aria1th 3afe99d27e Fix resuming, Add options 2022-11-27 10:57:43 +09:00
aria1th 0daf9874d0 FIx Scheduler 2022-11-27 10:23:43 +09:00
aria1th daa2ee237a amplify image, correct annealing, show correct lr 2022-11-26 10:31:37 +09:00
aria1th 40fd21ea8d apply max steps when using beta scheduler 2022-11-25 14:19:10 +09:00
aria1th 730320e65f Write float instead of list 2022-11-24 17:45:10 +09:00
aria1th 4af478f504 fix GradScaler 2022-11-24 14:15:38 +09:00
AngelBottomless 087040c9f8
fix loss save and CUDA memory 2022-11-24 13:17:15 +09:00
AngelBottomless 3b54b9ca58
Add external pr 2022-11-23 21:33:31 +09:00