the code logic in the #371 line of channel_pruning_env.py is missing one line code: self.shared_idx.append(share_group) should be added after the for loop.