Yes, this should work. Do you see any issues or is this just a general question?
You wouldn’t see the last batch norm layer (self.bn_fc1) at the end of the model usually, but it might fit your use case. I would nevertheless compare it to a model without this layer and check, which one works better.