pix2pix training in ml5.js