Outstanding TODO is to verify that the shapes of things are as expected (they are currently, as I've checked by printing out their shapes). The gramian and loss calculations are a little bit complicated, though I don't think they're too bad.
Additionally need to change to batch norm on the conv blocks.
Outstanding TODO is to verify that the shapes of things are as expected (they are currently, as I've checked by printing out their shapes). The gramian and loss calculations are a little bit complicated, though I don't think they're too bad.
Additionally need to change to batch norm on the conv blocks.