def resize_()

in models/densenet_efficient_multi_gpu.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

    def resize_(self, size):
        for device_no, sto in enumerate(self.multi_storage):
            if sto.size() < size:
                with torch.cuda.device(device_no): # this line is crucial!!
                    sto.resize_(size)
        return self