Skip to content
This repository was archived by the owner on Apr 23, 2025. It is now read-only.

Commit e3b8a6b

Browse files
authored
Make 'GoModel' stored properties be variables. (#174)
New 'Differentiable' derived conformances will not include constant properties in the tangent space.
1 parent b9920ae commit e3b8a6b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

MiniGo/Models/GoModel.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,9 @@ extension ResidualIdentityBlock: LoadableFromPythonCheckpoint {
106106

107107
// This is needed because we can't conform tuples to protocols
108108
public struct GoModelOutput: Differentiable {
109-
public let policy: Tensor<Float>
110-
public let value: Tensor<Float>
111-
public let logits: Tensor<Float>
109+
public var policy: Tensor<Float>
110+
public var value: Tensor<Float>
111+
public var logits: Tensor<Float>
112112
}
113113

114114
public struct GoModel: Layer {

0 commit comments

Comments
 (0)