Skip to content

Commit 24c88cf

Browse files
committed
Fixed save_path keyword.
1 parent 6a9f99c commit 24c88cf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Pilot3/P3B5/p3b5_darts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def train(trainloader, validloader, model, architecture, criterion, optimizer, l
6969
print(f'Step: {step} loss: {meter.loss_meter.avg:.4}')
7070

7171
meter.update_epoch()
72-
meter.save(args.savepath)
72+
meter.save(args.save_path)
7373

7474

7575
def infer(validloader, model, criterion, args, tasks, device, meter):
@@ -95,7 +95,7 @@ def infer(validloader, model, criterion, args, tasks, device, meter):
9595
print(f'>> Validation: {step} loss: {meter.loss_meter.avg:.4}')
9696

9797
meter.update_epoch()
98-
meter.save(args.savepath)
98+
meter.save(args.save_path)
9999

100100
return meter.loss_meter.avg
101101

0 commit comments

Comments
 (0)