aist9 / autoencoder

MIT License
1 stars 0 forks source link

featureのやつについて #4

Closed YanoSanso closed 5 years ago

YanoSanso commented 5 years ago

・train_ae が trainae になっており、train_stacked での呼び出しと一致していない。 ・AEの活性化関数の条件分岐の中で relu が sigmoidになっている ・gpu_device の引数を train_stacked に追加する。 ・活性化関数を層ごとに決められるようにするほうがよさそう。