

以下のようなCNNをpythonで作成したのですが、計算を回したところlossの値が変化しませんでした。  
原因がわからないのですが、教えていただけないでしょうか？

[code]

    from torch.nn import MSELoss
    device = 'cuda'
    model = Model().to(device)
    criterion = MSELoss()
    optimizer = Adam(model.parameters(), lr=0.003)
    
    for epoch in tqdm(range(100)):
     print('epoch:', epoch + 1)
     for (inputs, labels) in tqdm(dataloader):
       inputs, labels = inputs.float().to(device), labels.float().to(device)
       optimizer.zero_grad()
       start_prediction = model(inputs)
       loss = criterion(start_prediction, labels)
       loss.backward()
       optimizer.step()
    
     for (inputs, labels) in tqdm(validloader):
       inputs, labels = inputs.float().to(device), labels.float().to(device)
       start_prediction = model(inputs)
       loss = criterion(start_prediction, labels)
    
    
[/code]


CC BY-SA 4.0
2020-10-03T10:25:04.997
70898
2020-10-03T10:25:04.997
42166
question
python 機械学習 pytorch ニューラルネットワーク
pytorchで作ったCNNが学習しません