Improve documentation

borglab · Nov 26, 2020 · f3e371f · f3e371f
1 parent 171eed7
commit f3e371f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Sources/SwiftFusion/Inference/IdentityProjection.swift b/Sources/SwiftFusion/Inference/IdentityProjection.swift
@@ -14,11 +14,9 @@
 import TensorFlow
 import PenguinStructures
 
-/// A class performing common activities in the IdentityProjection framework.
+/// A class that does a identity feature mapping (directly outputs the pixels flattened)
 ///
-/// NOTE: This class can do normalization
-///
-/// - Input shape for training: [N, H, W, C]
+/// - Input shape for initialization: [N, H, W, C]
 /// - W matrix: [feature, H*W*C]
 /// - Output: [feature]
 public struct IdentityProjection {