在PyTorch中,通常情况下,模型的参数没有严格的限制,但如果我想让它们保持在[0,1]范围内呢?有没有一种方式可以阻止参数更新到该范围之外?
在PyTorch中,通常情况下,模型的参数没有严格的限制,但如果我想让它们保持在[0,1]范围内呢?有没有一种方式可以阻止参数更新到该范围之外?
model = YourPyTorchModule()
for _ in range(epochs):
loss = ...
optimizer.step()
for p in model.parameters():
p.data.clamp_(-1.0, 1.0)