Skip to content

Commit d0ecde6

Browse files
committed
Adjust number of queries.
1 parent 4cd6018 commit d0ecde6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/ParallelReverseAutoDiff.Test/GraphAttentionPaths/GraphAttentionPathsNeuralNetwork.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ public async Task Initialize()
107107

108108
for (int i = 0; i < 7; ++i)
109109
{
110-
var model = new TransformerNeuralNetwork(this.numLayers, this.numQueries, i + 2, this.numFeatures * (int)Math.Pow(2d, (double)this.numLayers), i + 2, this.learningRate, this.clipValue);
110+
var model = new TransformerNeuralNetwork(this.numLayers, this.numQueries / 2, i + 2, this.numFeatures * (int)Math.Pow(2d, (double)this.numLayers), i + 2, this.learningRate, this.clipValue);
111111
this.transformerNeuralNetwork.Add(model);
112112
await this.transformerNeuralNetwork[i].Initialize();
113113
this.modelLayers = this.modelLayers.Concat(this.transformerNeuralNetwork[i].ModelLayers).ToList();

0 commit comments

Comments
 (0)