Skip to content

Instantly share code, notes, and snippets.

@ModMaamari
Created July 12, 2020 20:11
Show Gist options
  • Save ModMaamari/86f0339148252123d020df8d29f62cf8 to your computer and use it in GitHub Desktop.
Save ModMaamari/86f0339148252123d020df8d29f62cf8 to your computer and use it in GitHub Desktop.
# Fit on all samples as one batch, log only on terminal state
self.model.fit(x = np.array(X).reshape(-1, *env.ENVIRONMENT_SHAPE),
y = np.array(y),
batch_size = MINIBATCH_SIZE, verbose = 0,
shuffle=False, callbacks=[self.tensorboard] if terminal_state else None)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment