andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Narrow hyperparams

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-04 07:50:02 +01:00
parent 6c2a20c73b
commit cb8a791477
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
2 changed files with 85 additions and 251 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/editor/models/__init__.py
View file

@ -15,13 +15,13 @@ MODELS = {
# "v1": v1,
"SimpleCNN": SimpleCNN,
"Residual": Residual,
"NormalisedCNN": NormalisedCNN,
"SmartRes": SmartRes,
# "NormalisedCNN": NormalisedCNN,
# "SmartRes": SmartRes,
# "AttentionNet": AttentionNet,
"attention2": EnhancedAestheticHistogramNet,
"advanced_attention": advanced_attention,
"Res2": Res2,
"attention1": PhotoEnhanceNetAdvanced,
# "attention2": EnhancedAestheticHistogramNet,
# "advanced_attention": advanced_attention,
# "Res2": Res2,
# "attention1": PhotoEnhanceNetAdvanced,
}