Cool, I've tried a few but settled on using EraX NSFW to do the same.

just wondering, why that one? it looks small, which is probably doesn't require a gpu.

The dataset seems to be images of high production value (e.g. limited races, staged poses, etc). If I have time, I will compare it with Bumble's model, but I think the images I'm trying to identify are closer to Bumble's training set.