From 2a876b381352ecffa0e4ea76c8ed57eea7354cfc Mon Sep 17 00:00:00 2001
From: araki <araki@qoncept.co.jp>
Date: Wed, 24 Jul 2019 10:53:40 +0900
Subject: [PATCH] Replace Autoencoder's tanh activation with sigmoid

---
 Autoencoder/main.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Autoencoder/main.swift b/Autoencoder/main.swift
index 0d69e0925f4..2dee45332d6 100644
--- a/Autoencoder/main.swift
+++ b/Autoencoder/main.swift
@@ -91,7 +91,7 @@ struct Autoencoder: Layer {
     var decoder2 = Dense<Float>(inputSize: 12, outputSize: 64, activation: relu)
     var decoder3 = Dense<Float>(inputSize: 64, outputSize: 128, activation: relu)
     var decoder4 = Dense<Float>(inputSize: 128, outputSize: imageHeight * imageWidth,
-        activation: tanh)
+        activation: sigmoid)
 
     @differentiable
     func callAsFunction(_ input: Input) -> Output {