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)