pytorch
zjy2015302395
ACM要AC
展开
-
pytorch入门(3)
语言模型lstmimport torchtextfrom torchtext.vocab import Vectorsimport torchimport numpy as npimport randomBATCH_SIZE = 32EMBEDDING_SIZE = 100MAX_VOCAB_SIZE = 5000import torch.nn as nnclass RNNModel(nn.Module): def __init__(self, rnn_type,vocab_原创 2020-05-23 13:49:02 · 254 阅读 · 0 评论 -
pytorch入门(2)
fizzbuzz游戏3倍数fizz 5倍数buzz 15倍数fizzbuzz。其余数字输出本身12fizz4buzzfizz78fizzbuzz11fizz1314fizzbuzzdef fizz_buzz_encode(i): if i %15 == 0 : return 3 elif i%5 == 0 : return 2 elif i %3 == 0: return 1 else: return 0def fizz_buzz_de原创 2020-05-20 17:05:00 · 484 阅读 · 0 评论 -
pytorch入门(1)
import torchimport numpy as npN, D_in, H, D_out = 64,1000,100,10x = torch.randn(N,D_in)y = torch.randn(N,D_out)w1 = torch.randn(D_in, H,requires_grad = True)w2 = torch.randn(H,D_out,requires_grad = True)learning_rate = 1e-6for t in range(500):原创 2020-05-20 00:50:29 · 341 阅读 · 0 评论