dzysunshine的博客

预训练模型总结

基于Anaconda下的pytorch的学习，具体安装课参考Anaconda的安装和环境使用，有详细介绍。附上pytorch官网链接：https://pytorch.org/docs/stable/index.html下面直接开始1. 查看安装pytorch的相关信息import torchimport torch.nn as nnimport torchvisionprint(torch.__version__)print(torch.version.cuda)print(torch.b

pytorch相关用法设置随机种子，复现模型结果当我们想复现模型结果的时候，需要将所有的种子全部固定下来，可以实现多次运行结果一致。def set_rand_seed(seed=1): print("Random Seed: ", seed) random.seed(seed) np.random.seed(seed) # CPU随机种子 torch.manual_seed(seed) # GPU随机种子 torch.cuda.manual_se