from torch import nn
from torch.nn import functional as F
+import matplotlib.pyplot as plt
+
######################################################################
-parser = argparse.ArgumentParser(description='Toy RNN.')
+parser = argparse.ArgumentParser(description='Toy attention model.')
parser.add_argument('--nb_epochs',
type = int, default = 250)
######################################################################
-import matplotlib.pyplot as plt
-import matplotlib.collections as mc
-
def save_sequence_images(filename, sequences, tr = None, bx = None):
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)