zc/mlp-project
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

FIX: deprecated loss reduction argument

Browse source
This commit is contained in:
Zhengyi Chen 2024-03-03 22:39:30 +00:00
parent c4905acf6d
commit da8287b7e8
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
train.py
View file

@ -148,7 +148,7 @@ def worker(rank: int, args: Namespace):
model = model.cuda() model = model.cuda()
# criterion, optimizer, scheduler # criterion, optimizer, scheduler
criterion = nn.L1Loss(size_average=False) criterion = nn.L1Loss(reduction="sum")
if device is not None: if device is not None:
criterion = criterion.to(device) criterion = criterion.to(device)
elif torch.cuda.is_available(): elif torch.cuda.is_available():