DrJonnyT

"""Script to compare the tensorflow and pytorch implementations of neuralprocesses.
The training losses printed at the end should be very similar, the differences are 
thought to be due to different random seeds.
This script currently works for gnp, agnp, and convgnp architectures (defined
on L93 & 94).
"""

import tensorflow as tf
import torch
import numpy as np