def test_default_memory_limit(self): utils.load_npy(self.npy_file) assert utils.get_memory_limit() == self.default_memory_limit memory_limit = utils.get_memory_limit() utils.set_memory_limit(1) utils.load_npy(self.npy_file) utils.set_memory_limit(memory_limit)
out = self.layer2(out) out = self.layer3(out) out = self.layer4(out) out = self.layer5(out) out = self.layer6(out) out = self.layer7(out) out = self.layer8(out) out = self.layer9(out) out = out.view(out.size(0), -1) out = self.fc(out) return out if __name__ == '__main__': set_memory_limit(30 * 1024**3) # Preprocessing waveform_transform = transforms.Compose([ transforms.ZeroMean(), transforms.SoftClip(1e-4), transforms.ToTensor(), ]) target_transform = transforms.Compose([ transforms.Select(0), transforms.ToInt({ 'N': 0, 'P': 1, 'S': 2, }),
def test_set_memory_limit(self): default_memory_limit = utils.get_memory_limit() utils.set_memory_limit(1) assert utils.get_memory_limit() == 1 utils.set_memory_limit(default_memory_limit)