Tests I need

data constructors

verify that the metatrainer constructor is being initialized correctly

data splitting tests

verify splitting of data into tasks

verify shuffling of batches from tasks

Inner loop tests

outer loop tests

Reptile tests

Verify model recreation using deep copy

Verify linear update

Maml tests

Verify the meta loss is calculated correctly

Verify that the meta loss is calculated correctly when stacking multiple loss tensors

Verify optimizers are not changing when they're not supposed to change

Validation tests

Verify that validation is working correctly