I believe I eventually decided pytorch didn’t like the concept of pickling torch.autograd.functions.  I ended up making a function that saved the variables from the class, deleted them all then pickled and upon load created new class objects and filled in the values.