1. ホーム
  2. python

[解決済み】pytorchでmodel.eval()は何をするのですか?

2022-02-05 05:57:10

質問

を使っています。 このコード と表示されました。 .eval()

model.eval()

.eval()

torch.no_grad()

model.eval()

# evaluate model:
model.eval()

with torch.no_grad():
    ...
    out_data = model(data)
    ...

training

# training step
...
model.train()
...