Attention Head Pruning for Model Compression - Coding problem | OneNoughtOne