Commit Graph

45 Commits

Author SHA1 Message Date
Falko Victor Habel 619e17c32c going with batchsize 1 2025-02-15 13:17:17 +01:00
Falko Victor Habel e69d0e90ec fixed checkpoints tensors 2025-02-15 13:08:09 +01:00
Falko Victor Habel 75be3291d3 added custom upscaler model 2025-02-14 22:39:20 +01:00
Falko Victor Habel 1234fc5beb added device type to autocast 2025-02-14 22:27:19 +01:00
Falko Victor Habel ca29ee748b fixed rsults 2025-02-14 22:23:38 +01:00
Falko Victor Habel 9645e1da23 fixed checkpoint 2025-02-14 22:14:09 +01:00
Falko Victor Habel 5a6680178a try checkpoints 2025-02-14 22:10:07 +01:00
Falko Victor Habel e7e7e96001 max gpu usage added 2025-02-14 22:03:04 +01:00
Falko Victor Habel ca44dd8a77 added torch.amp support 2025-02-14 21:48:52 +01:00
Falko Victor Habel b4dd550f8d downsized params 2025-02-14 21:33:47 +01:00
Falko Victor Habel 19e5b72724 simplified code 2025-02-12 06:59:18 +01:00
Falko Victor Habel 8fafbebe45 fixed loading 2025-02-08 16:55:54 +01:00
Falko Victor Habel 20dd9f68ed added tqdm to improve code 2025-02-08 12:52:48 +01:00
Falko Victor Habel a48bad5a49 limit dataset 2025-02-07 15:26:57 +01:00
Falko Victor Habel 39efdff09e updated datasets 2025-02-06 20:54:05 +01:00
Falko Victor Habel bac4a9010a updated dataset loading 2025-02-06 18:33:43 +01:00
Falko Victor Habel 72a5959bc1 updated batchsize and fixed key names 2025-02-05 22:32:39 +01:00
Falko Victor Habel 1a66169b37 updated keys 2025-02-05 21:33:33 +01:00
Falko Victor Habel e298ac4c45 manually loading the correct modle again 2025-02-05 21:27:42 +01:00
Falko Victor Habel de6a67cb4e updated finetuning script 2025-02-05 16:16:31 +01:00
Falko Victor Habel bb22d0a6da added decoder models fits 2025-01-31 17:34:33 +01:00
Falko Victor Habel be74658ceb added decoeder model 2025-01-31 17:30:41 +01:00
Falko Victor Habel 9187ebe012 debug print 2025-01-31 17:14:28 +01:00
Falko Victor Habel 325faef9d5 push 2025-01-31 17:02:35 +01:00
Falko Victor Habel 50fa103579 printing 2025-01-31 16:57:13 +01:00
Falko Victor Habel 804882d7a2 corrected dataset paths 2025-01-31 16:11:21 +01:00
Falko Victor Habel 0b04ebab7a corected path again 2025-01-31 16:06:12 +01:00
Falko Victor Habel 6b35463027 upgaeed path for base model 2025-01-31 16:04:54 +01:00
Falko Victor Habel 13fb2b76c1 updated finetuning 2025-01-31 16:03:28 +01:00
Falko Victor Habel 9ec563e86d updated Finetuner to correctly log losses 2025-01-30 20:49:34 +01:00
Falko Victor Habel 8c49cc7e01 fixed tqdm in finetuning 2025-01-30 13:51:49 +01:00
Falko Victor Habel 0484ae01b1 added tqdm and removed doubles 2025-01-30 13:21:56 +01:00
Falko Victor Habel 34c547fb23 next try 2025-01-30 13:14:45 +01:00
Falko Victor Habel 9b7a182782 debug mode 2025-01-30 12:52:57 +01:00
Falko Victor Habel 4037a07764 updated process 2025-01-30 12:39:24 +01:00
Falko Victor Habel be5bb53620 working transform 2025-01-30 11:56:17 +01:00
Falko Victor Habel 1f33f22bea convert the bytes first 2025-01-30 10:45:18 +01:00
Falko Victor Habel 73d52f733c updated config for updated model 2025-01-30 10:41:37 +01:00
Falko Victor Habel 2121316e3b finetune improvement 2025-01-30 10:36:15 +01:00
Falko Victor Habel 4a60045320 first finetune try 2025-01-29 23:06:49 +01:00
Falko Victor Habel 914d002602 added new aiiun script with first draft for pip project 2025-01-29 22:30:57 +01:00
Falko Victor Habel 71da7ed2f1 removed old scripts 2025-01-29 22:30:35 +01:00
Falko Victor Habel d85faadcc1 first finetune script 2025-01-29 19:26:32 +01:00
Falko Victor Habel 2f14301550 added init files for all different aiunn versions 2025-01-07 09:23:31 +01:00
Falko Victor Habel feea191388 Initial commit 2024-12-30 22:41:40 +00:00