Minor fixes

FangjinhuaWang · Sep 10, 2021 · 88d0bf0 · 88d0bf0
1 parent c0d7ff8
commit 88d0bf0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,3 @@
-.idea
-venv
-__pycache__
+.idea
+venv
+__pycache__
diff --git a/models/patchmatch.py b/models/patchmatch.py
@@ -377,7 +377,6 @@ def __init__(
         self.evaluation = Evaluation(self.G, self.stage, self.evaluate_neighbors, self.patchmatch_iteration)
         # adaptive propagation
         # last iteration on stage 1 does not have propagation, but we still define this for TorchScript export compatibility
-        # This may have some impact on training! need to investigate
         self.propa_conv = nn.Conv2d(
             in_channels=self.propa_num_feature,
             out_channels=max(2 * self.propagate_neighbors, 1),