 Falko Victor Habel
|
3cb2a18ad9
|
updated version to support proper saving when pretraining
|
2025-02-27 18:56:42 +01:00 |
|
Falko Victor Habel
|
1faf34749a
|
added serilatsation
|
2025-02-24 15:16:46 +01:00 |
|
Falko Victor Habel
|
c9b6a8926b
|
updated readme
|
2025-02-24 14:17:25 +01:00 |
|
Falko Victor Habel
|
50e91b10e8
|
fixed model loading due to a bug
|
2025-02-24 14:13:10 +01:00 |
|
Falko Victor Habel
|
b1c486afee
|
added fp16 and bf16 support when loading model
|
2025-02-24 13:41:11 +01:00 |
|
Falko Victor Habel
|
f8e59c5896
|
added cpu support when loading the model
|
2025-01-31 09:13:58 +01:00 |
|
Falko Victor Habel
|
3c0e9e8ac1
|
fixed savings
|
2025-01-28 17:18:21 +01:00 |
|
Falko Victor Habel
|
3631df7f0a
|
updated pretrainer to handle multiple classes and configs.
|
2025-01-28 11:42:03 +01:00 |
|
Falko Victor Habel
|
a369c49f15
|
improved pretraining
|
2025-01-28 11:27:42 +01:00 |
|
Falko Victor Habel
|
7de7eef081
|
updated pretraing to create a extra class for Pretraining
|
2025-01-28 11:16:09 +01:00 |
|
Falko Victor Habel
|
1e665c4604
|
added first pip install version 0.1
|
2025-01-28 10:58:33 +01:00 |
|
Falko Victor Habel
|
102d952043
|
increased epochs
|
2025-01-27 11:06:42 +01:00 |
|
Falko Victor Habel
|
8d08bfc14c
|
removed rint statemetns and added csv saving
|
2025-01-27 10:56:16 +01:00 |
|
Falko Victor Habel
|
58baf0ad3c
|
overall improvement
|
2025-01-27 10:43:59 +01:00 |
|
Falko Victor Habel
|
fe4d6b5b22
|
corrected viewing and some prints
|
2025-01-27 10:26:28 +01:00 |
|
Falko Victor Habel
|
b6b63851ca
|
addeed tasks for both denosing and rotation
|
2025-01-27 10:15:00 +01:00 |
|
Falko Victor Habel
|
13cf1897ae
|
handle both pretraining methods
|
2025-01-27 09:32:20 +01:00 |
|
Falko Victor Habel
|
91a568731f
|
removed placeholder collate function
|
2025-01-27 09:26:08 +01:00 |
|
Falko Victor Habel
|
6c146f2767
|
added progressbar for batches
|
2025-01-27 09:13:22 +01:00 |
|
Falko Victor Habel
|
b546f4ee27
|
furhter batchsizing downsizing
|
2025-01-27 09:08:39 +01:00 |
|
Falko Victor Habel
|
8dad1d7150
|
downszied batchsize
|
2025-01-27 09:02:22 +01:00 |
|
Falko Victor Habel
|
32526c3c30
|
5k image
|
2025-01-27 08:53:01 +01:00 |
|
Falko Victor Habel
|
d205346741
|
downsized trainingdata from 20k to 1k
|
2025-01-27 08:44:46 +01:00 |
|
Falko Victor Habel
|
3749ba9c5f
|
updated base models MaxPool2D
|
2025-01-27 08:39:42 +01:00 |
|
Falko Victor Habel
|
29f0d86ff7
|
kernel size as large as channel size
|
2025-01-26 23:26:42 +01:00 |
|
Falko Victor Habel
|
b501ae8317
|
new pretraining script
|
2025-01-26 23:20:15 +01:00 |
|
Falko Victor Habel
|
2b55f02b50
|
eorr handling because we have a tensor misshaping
|
2025-01-26 23:08:56 +01:00 |
|
Falko Victor Habel
|
8a809269e5
|
fix inline error
|
2025-01-26 23:04:49 +01:00 |
|
Falko Victor Habel
|
7c4aef0978
|
updated dataloader to work with tupels
|
2025-01-26 22:48:29 +01:00 |
|
Falko Victor Habel
|
3f6e6514a9
|
rgba conversion to rgb
|
2025-01-26 22:17:37 +01:00 |
|
Falko Victor Habel
|
cae3fa7fb3
|
proper image transformation
|
2025-01-26 22:08:59 +01:00 |
|
Falko Victor Habel
|
7a1eb8bd30
|
updated loader
|
2025-01-26 21:58:13 +01:00 |
|
Falko Victor Habel
|
a8cd9b00e5
|
limit data loadng to 10k
|
2025-01-26 18:41:14 +01:00 |
|
Falko Victor Habel
|
e0abdb9d39
|
fixed datasplitting
|
2025-01-26 18:40:19 +01:00 |
|
Falko Victor Habel
|
51da6b5aa2
|
corrected loading
|
2025-01-26 18:19:54 +01:00 |
|
Falko Victor Habel
|
b5da2e477d
|
correct allogcation for indexes
|
2025-01-26 18:15:05 +01:00 |
|
Falko Victor Habel
|
9124221346
|
correct loading files
|
2025-01-26 17:26:13 +01:00 |
|
Falko Victor Habel
|
b7dc835a86
|
fixed dataloading
|
2025-01-26 16:20:33 +01:00 |
|
Falko Victor Habel
|
00168af32d
|
corrected init
|
2025-01-26 13:48:52 +01:00 |
|
Falko Victor Habel
|
338ac5dee5
|
corrected imports
|
2025-01-26 13:26:01 +01:00 |
|
Falko Victor Habel
|
59b2784e92
|
fixed spelling error
|
2025-01-26 13:10:24 +01:00 |
|
Falko Victor Habel
|
e5a5618160
|
correct copying
|
2025-01-26 13:09:23 +01:00 |
|
Falko Victor Habel
|
32daaadddd
|
updated pretrain method
|
2025-01-26 13:05:24 +01:00 |
|
Falko Victor Habel
|
de3d58f6db
|
improved cnn
|
2025-01-24 18:49:25 +01:00 |
|
Falko Victor Habel
|
8ac31c5bf1
|
improved shared model to have ~10% of params
|
2025-01-24 18:23:54 +01:00 |
|
Falko Victor Habel
|
599b8c4835
|
working shared model (with way to few params)
|
2025-01-24 18:04:44 +01:00 |
|
Falko Victor Habel
|
6e6f4c4a21
|
updated models and config to improve parameter handling and adding a copy function to use the same base config for mutliple models
|
2025-01-22 14:23:03 +01:00 |
|
Falko Victor Habel
|
ab58d352c4
|
updated saving and first implementation of new additonal parameter handling
|
2025-01-22 14:16:56 +01:00 |
|
Falko Victor Habel
|
26b701fd77
|
corrected activation function
|
2025-01-22 11:20:05 +01:00 |
|
Falko Victor Habel
|
74973a325b
|
updated models for improved config
|
2025-01-22 11:19:55 +01:00 |