Merge pull request #7 from StephanEggermont/patch-1
Browse files- models/ctm.py +1 -1
models/ctm.py
CHANGED
|
@@ -247,7 +247,7 @@ class ContinuousThoughtMachine(nn.Module):
|
|
| 247 |
|
| 248 |
def set_initial_rgb(self):
|
| 249 |
"""
|
| 250 |
-
This is largely to accommodate training on
|
| 251 |
doesn't hurt the model in any way that we can tell.
|
| 252 |
"""
|
| 253 |
if 'resnet' in self.backbone_type:
|
|
|
|
| 247 |
|
| 248 |
def set_initial_rgb(self):
|
| 249 |
"""
|
| 250 |
+
This is largely to accommodate training on grayscale images and is legacy, but it
|
| 251 |
doesn't hurt the model in any way that we can tell.
|
| 252 |
"""
|
| 253 |
if 'resnet' in self.backbone_type:
|