That looks like a reasonable way to initialize it, though your model is not exactly the same as AlexNet both in the layer definitions and the fully connected layers.
There is also torchvision which has a pre-setup version of the model - https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py 134