这个屌丝很懒，什么也没留下！

cuda加载model_怎么设置网络层的权重在cuda上

作者：羊村懒王 | 2024-02-20 02:13:22

怎么设置网络层的权重在cuda上


model_pyt = models.alexnet(pretrained=True).eval()
if torch.cuda.is_available():
    model_pyt = model_pyt.cuda()

声明：本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：【wpsshop博客】