比較検索なら 天秤AI byGMO
システム開発アシスタント

システム開発アシスタント

言語やフレームワークを自由に指定することで、システム開発アシスタントとして何でも質問に答えてくれます。

5
0
136
33
user_123038585507129344
user_108961360132238336
user_67825708779517952
がお気に入り登録しました。

投稿日時:

  • プロンプト実行例
  • プロンプトを見る
システム開発アシスタント

質問したい言語の名前を入力してください。また、フレームワークやその他の制約条件を追加することもできます。

言語名

フレームワーク

その他の制約条件

システム開発アシスタント

Torchを使用していくつかの質問に答えます。どうかお知らせください。

  1. PyTorchで新しいニューラルネットワークを定義する方法はありますか?
  2. データセットからのバッチデータの取得方法を教えていただけますか?
  3. ニューラルネットワークの学習と推論におけるGPUの利用方法を教えてください。

MNISTデータの学習をしたいです。

システム開発アシスタント

に、PyTorchを使用してMNISTデータセットの学習を行うためのコードを示。PyTorchのtorchvisionモジュールからMNISTデータセットをダウンロードし、ニューラルネットワークを定義し、学習を行います。

Python
1import torch
2import torchvision
3import torchvision.transforms as transforms
4import torch.nn as nn
5import torch.optim as optim
6
7# データセットの読み込みと前処理
8transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))])
9trainset = torchvision.datasets.MNIST(root='./data', train=True, download=True, transform=transform)
10trainloader = torch.utils.data.DataLoader(trainset, batch_size=4, shuffle=True, num_workers=2)
11
12# ニューラルネットワークの定義
13class Net(nn.Module):
14    def __init__(self):
15        super(Net, self).__init__()
16        self.conv1 = nn.Conv2d(1, 6, 5)
17        self.pool = nn.MaxPool2d(2, 2)
18        self.conv2 = nn.Conv2d(6, 16, 5)
19        self.fc1 = nn.Linear(16 * 4 * 4, 120)
20        self.fc2 = nn.Linear(120, 84)
21        self.fc3 = nn.Linear(84, 10)
22
23    def forward(self, x):
24        x = self.pool(F.relu(self.conv1(x)))
25        x = self.pool(F.relu(self.conv2(x)))
26        x = x.view(-1, 16 * 4 * 4)
27        x = F.relu(self.fc1(x))
28        x = F.relu(self.fc2(x))
29        x = self.fc3(x)
30        return x
31
32net = Net()
33
34# 損失関数と最適化アルゴリズムの定義
35criterion = nn.CrossEntropyLoss()
36optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
37
38# ニューラルネットワークの学習
39for epoch in range(2):
40    running_loss = 0.0
41    for i, data in enumerate(trainloader, 0):
42        inputs, labels = data
43        optimizer.zero_grad()
44        outputs = net(inputs)
45        loss = criterion(outputs, labels)
46        loss.backward()
47        optimizer.step()
48        running_loss += loss.item()
49        if i % 2000 == 1999:    # 2000 ミニバッチごとに表示
50            print(f'[{epoch + 1}, {i + 1}] loss: {running_loss / 2000:.3f}')
51            running_loss = 0.0
52
53print('学習が完了しました。')

このコードでは、torchvisionを使用してMNISTデータセットをダウンロードし、torch.utils.data.DataLoaderを使用してデータをバッチで取得しています。また、Netクラスを定義してニューラルネットワークを作成し、最適化手法としてSGDを使用しています。学習ループ内でミニバッチごとに損失を計算し、モデルを最適化しています。

コメント 0

他のプロンプトもチェック

  • プロンプト生成

    ChatGPTなどで使う通常のプロンプトを、Claudeに最適なXMLタグを使ったプロンプトに変換します。実は、XMLタグを使用することでClaudeの性能が向上すると公式がアナウンスしているんです✨ ✅直接プロンプトを入れるだけで速攻変換してくれます! (もし上手くいかない場合は何度か再生成してください。)
    14
    0
    37
    434
  • チャットボット

    推奨モデル - GPT-4o
    あなたの「隠れコミュ力」を診断する。AIとの会話を3回程度するだけ。100点満点で採点。社会でうまくやっていくなら60点はほしい。
  • マーケティング一般

    推奨モデル - Claude
    最初はHikakuになってますが、Sonnet推奨。Twitterのネタツイートを生成します!
  • GPTs

    キャラクター作成

    花押クリエイターは、日本の伝統的な署名や印「花押」のデザインをサポートするためのツールです。国内外のさまざまな文化背景を考慮し、名前やシンボルを基にした個性的で芸術的な花押デザインを提案します。 このツールは、歴史的な作法や文化的背景を尊重しながら、現代的なアプローチにも対応します。作成した花押のコンセプトや意図を分かりやすく説明し、カスタマイズや調整も簡単に行うことが出来ます。 初めての方から花押愛好家まで、どなたでもご利用いただける「花押クリエイター」で、あなただけの特別なデザインを作りませんか? #画像作成 #花押 #サイン
    6
    1
    739
    1.63K