深度学习||常用代码块||记录不定期更新

if __name__ == '__main__':
# 在某个文件下测试

input = torch.randn(1, 3, 20, 20, 16)
net = Net()
print("params: ", sum(p.numel() for p in net.parameters()))
out = net(input)
print(out.shape)

# 在调整网络结构时候经常用得到的一个

if not os.path.exists('./dataset'):
    os.mkdir('./dataset')

# 如果当前目录不存在，就创建

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
print(torch.cuda.get_device_name(0))

# 看包的版本

np.random.seed(3407)
torch.manual_seed(3407)
torch.cuda.manual_seed_all(3407)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False

# 设置随机种子