在程序中输出临时文件大小
1
2
3
4
5
6
def print_size_of_model(model):
torch.save(model.state_dict(), "temp.p")
print('Size (MB):', os.path.getsize("temp.p")/2**20)
os.remove('temp.p')
print_size_of_model(model)
Python常用代码段
This post is licensed under CC BY 4.0 by the author.
Recently Updated
Contents