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

Commit 2a8ca19

Browse files
committed
Drop <Float> type parameters which can be inferred
1 parent a723379 commit 2a8ca19

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Examples/LeNet-MNIST/main.swift

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ let dataset = MNIST(batchSize: batchSize)
2323
// The LeNet-5 model, equivalent to `LeNet` in `ImageClassificationModels`.
2424

2525
let denseDef = AutoDenseDefinition<Float>(outputSize: 120, activation: relu)
26-
.then(AutoDenseDefinition<Float>(outputSize: 84, activation: relu))
27-
.then(AutoDenseDefinition<Float>(outputSize: 10))
26+
.then(AutoDenseDefinition(outputSize: 84, activation: relu))
27+
.then(AutoDenseDefinition(outputSize: 10))
2828

2929
var classifier = Sequential {
3030
Conv2D<Float>(filterShape: (5, 5, 1, 6), padding: .same, activation: relu)

0 commit comments

Comments
 (0)