Thanks. Oh, and I also found someone who did the Titanic dataset using Fast.ai, and since I’ve done that one using sklearn before, that might hopefully bridge this knowledge gap.
Also will take into your suggestion of calling nn.Module directly if I don’t need a very fancy model, at least to start.