A 320 meter base is rated 320 amps continuous or 400 amp max. This is no different from a standard 400 amp panel.
A 320 sockets is the right item to use for a 400 amp service as long as your local power company does not want CTs.
In my area we use plug in meters for services 400 amps and below.