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

Make 'GoModel' stored properties be variables. #174

Merged
merged 1 commit into from
Jun 26, 2019
Merged

Make 'GoModel' stored properties be variables. #174

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions MiniGo/Models/GoModel.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -106,9 +106,9 @@ extension ResidualIdentityBlock: LoadableFromPythonCheckpoint {

// This is needed because we can't conform tuples to protocols
public struct GoModelOutput: Differentiable {
public let policy: Tensor<Float>
public let value: Tensor<Float>
public let logits: Tensor<Float>
public var policy: Tensor<Float>
public var value: Tensor<Float>
public var logits: Tensor<Float>
}

public struct GoModel: Layer {
Expand Down