Skip to content

Commit babfa3a

Browse files
committed
FashionMNIST: use SELU instead of RELU
1 parent 8c0f45f commit babfa3a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

FashionMnistClassification/FashionMnistClassification.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
static class FashionMnistClassification {
1212
static void Main() {
13+
Console.Title = nameof(FashionMnistClassification);
1314
GradientLog.OutputWriter = Console.Out;
1415
GradientSetup.UseEnvironmentFromVariable();
1516

@@ -27,7 +28,7 @@ static void Main() {
2728
var model = new Sequential(new Layer[] {
2829
// will be able to do: new Flatten(kwargs: new { input_shape = (28, 28) }),
2930
new Flatten(kwargs: new PythonDict<string, object> { ["input_shape"] = (28, 28) }),
30-
new Dense(units: 128, activation: tf.nn.relu_fn),
31+
new Dense(units: 128, activation: tf.nn.selu_fn),
3132
new Dense(units: 10, activation: tf.nn.softmax_fn),
3233
});
3334

0 commit comments

Comments
 (0)