Skip to content

4xDRCT-mssim-pretrains

Compare
Choose a tag to compare
@Phhofm Phhofm released this 28 Apr 09:26
· 38 commits to main since this release
e9cdf82

Since no DRCT model releases exist yet, I am releasing my drct-s, drct and drct-l pretrains.
These can be used to speed up and stabilize early training stages when training new drct models.
Trained with mssim (and augs, and color and lumaloss) on downscaled nomosuni.
Training files are also provided additionally.

4xmssim_drct-s_pretrain

Scale: 4
Architecture: DRCT
Architecture Option: DRCT-S

Author: Philip Hofmann
License: CC-BY-0.4
Purpose: Pretrained
Subject: Anime, Photography
Input Type: Images
Release Date: 28.04.2024

Dataset: nomos_uni
Dataset Size: 2989
OTF (on the fly augmentations): No
Pretrained Model: None (=From Scratch)
Iterations: 75'000
Batch Size: 6
GT Size: 128-320

Description: A pretrain to start drct-s model training, since there are no official released drct pretrains yet I trained these myself and release them here.


4xmssim_drct_pretrain

Scale: 4
Architecture: DRCT
Architecture Option: DRCT

Author: Philip Hofmann
License: CC-BY-0.4
Purpose: Pretrained
Subject: Anime, Photography
Input Type: Images
Release Date: 28.04.2024

Dataset: nomos_uni
Dataset Size: 2989
OTF (on the fly augmentations): No
Pretrained Model: None (=From Scratch)
Iterations: 95'000
Batch Size: 2-6
GT Size: 128-320

Description: A pretrain to start drct model training, since there are no official released drct pretrains yet I trained these myself and release them here.


4xmssim_drct-l_pretrain

Scale: 4
Architecture: DRCT
Architecture Option: DRCT-L

Author: Philip Hofmann
License: CC-BY-0.4
Purpose: Pretrained
Subject: Anime, Photography
Input Type: Images
Release Date: 28.04.2024

Dataset: nomos_uni
Dataset Size: 2989
OTF (on the fly augmentations): No
Pretrained Model: None (=From Scratch)
Iterations: 108'000
Batch Size: 2-6
GT Size: 128-320

Description: A pretrain to start drct-l model training, since there are no official released drct pretrains yet I trained these myself and release them here.


Showcase:

slow.pics

Example1
Example2
Example3
Example4
Example5
Example6