Skip to content

Instantly share code, notes, and snippets.

@BenjaminWegener
BenjaminWegener / attention_layer_tfjs
Created April 14, 2021 18:02
Causal Attention Layer Tensorflow.js
//---------------------------------------------------
//setup
var dModel = 256; //dimension of embedding
var INPUTSIZE = 1024; //length of input
var BLOCKS = 3;
var LEARNING_RATE = 0.003;
var OPTIMIZER = tf.train.adam(LEARNING_RATE, beta_1 = 0.85, beta_2 = 0.9, epsilon = 1e-9);
var LOSS = 'categoricalCrossentropy';
var INITIALIZER = 'GlorotUniform';