You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Jan 7, 2025. It is now read-only.
Copy file name to clipboardExpand all lines: README.md
+30Lines changed: 30 additions & 0 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -33,6 +33,36 @@ ISR is compatible with Python 3.6 and is distributed under the Apache 2.0 licens
33
33
-[Maintainers](#maintainers)
34
34
-[License](#copyright)
35
35
36
+
## Troubleshooting
37
+
### Training not delivering good/patchy results
38
+
When training your own model, start with only PSNR loss (50+ epochs, depending on the dataset) and only then introduce GANS and feature loss. This can be controlled by the loss weights argument.
39
+
40
+
This is just sample, you will need to tune these parameters.
41
+
42
+
PSNR only:
43
+
```
44
+
loss_weights = {
45
+
'generator': 1.0,
46
+
'feature_extractor': 0.0,
47
+
'discriminator': 0.00
48
+
}
49
+
```
50
+
51
+
Later:
52
+
```
53
+
loss_weights = {
54
+
'generator': 0.0,
55
+
'feature_extractor': 0.0833,
56
+
'discriminator': 0.01
57
+
}
58
+
```
59
+
### Weights loading
60
+
If you are having trouble loading your own weights or the pre-trained weights (`AttributeError: 'str' object has no attribute 'decode'`), try:
0 commit comments