Clone/DIS
1
0
Fork 0
mirror of https://github.com/xuebinqin/DIS.git synced 2024-11-29 18:23:08 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

compute_loss

Browse Source
This commit is contained in:
Xuebin Qin 2022-07-17 11:00:12 -07:00
parent a802503bee
commit c1480958ac
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IS-Net/train_valid_inference_main.py
View File

@ -350,7 +350,7 @@ def train(net, optimizer, train_dataloaders, train_datasets, valid_dataloaders,
else: else:
# forward + backward + optimize # forward + backward + optimize
ds,_ = net(inputs_v) ds,_ = net(inputs_v)
loss2, loss = muti_loss_fusion(ds, labels_v) loss2, loss = net.compute_loss(ds, labels_v)
loss.backward() loss.backward()
optimizer.step() optimizer.step()