allow optionally specified architecture in pruning