iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updates slbm

Browse Source
This commit is contained in:
D-X-Y
2021-05-27 16:51:56 +00:00
parent 585ded31da
commit cec0cf993b
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/GeMOSA/baselines/slbm-ft.py
View File

@@ -131,7 +131,7 @@ def main(args):
# build optimizer
xmetric = ComposeMetric(metric_cls(True), SaveMetric())
future_x.to(args.device), future_y.to(args.device)
future_x, future_y = future_x.to(args.device), future_y.to(args.device)
future_y_hat = model(future_x)
future_loss = criterion(future_y_hat, future_y)
metric(future_y_hat, future_y)

2
exps/GeMOSA/baselines/slbm-nof.py
View File

@@ -130,7 +130,7 @@ def main(args):
# build optimizer
xmetric = ComposeMetric(metric_cls(True), SaveMetric())
future_x.to(args.device), future_y.to(args.device)
future_x, future_y = future_x.to(args.device), future_y.to(args.device)
future_y_hat = model(future_x)
future_loss = criterion(future_y_hat, future_y)
metric(future_y_hat, future_y)