Please keep off topic stuff in the Off Topic forum

[Data Sculpting]

Thanks, that helps. I ended up in about the same place using PyTorch. Worth rechecking your defaults though, because one bad assumption can make the rest of the thread look confusing.