Skip to content

Commit d032a9c

Browse files
committed
Remove loguru
We don't need to add this dependency just to print things.
1 parent 49cc6e5 commit d032a9c

File tree

2 files changed

+8
-11
lines changed

2 files changed

+8
-11
lines changed

examples/darts/advanced/example.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,6 @@
44
from torch.utils.data import DataLoader
55
from torchvision import datasets, transforms
66

7-
from loguru import logger
8-
97
import example_setup as bmk
108
import darts
119
import candle
@@ -91,10 +89,10 @@ def run(params):
9189

9290
scheduler.step()
9391
lr = scheduler.get_lr()[0]
94-
logger.info(f'\nEpoch: {epoch} lr: {lr}')
92+
print(f'\nEpoch: {epoch} lr: {lr}')
9593

9694
genotype = model.genotype()
97-
logger.info(f'Genotype: {genotype}\n')
95+
print(f'Genotype: {genotype}\n')
9896

9997
train(
10098
trainloader,
@@ -162,7 +160,7 @@ def train(trainloader,
162160
meter.update_batch_accuracy(prec1, batch_size)
163161

164162
if step % args.log_interval == 0:
165-
logger.info(f'Step: {step} loss: {meter.loss_meter.avg:.4}')
163+
print(f'Step: {step} loss: {meter.loss_meter.avg:.4}')
166164

167165
meter.update_epoch()
168166
meter.save(args.savepath)
@@ -187,7 +185,7 @@ def validate(validloader, model, criterion, args, tasks, meter, device):
187185
meter.update_batch_accuracy(prec1, batch_size)
188186

189187
if step % args.log_interval == 0:
190-
logger.info(f'>> Validation: {step} loss: {meter.loss_meter.avg:.4}')
188+
print(f'>> Validation: {step} loss: {meter.loss_meter.avg:.4}')
191189

192190
meter.update_epoch()
193191
meter.save(args.savepath)

examples/darts/uno/uno_example.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import torch.nn as nn
33
from torch import optim
44
from torch.utils.data import DataLoader
5-
from loguru import logger
65

76
import example_setup as bmk
87
import darts
@@ -74,10 +73,10 @@ def run(params):
7473

7574
scheduler.step()
7675
lr = scheduler.get_lr()[0]
77-
logger.info(f'\nEpoch: {epoch} lr: {lr}')
76+
print(f'\nEpoch: {epoch} lr: {lr}')
7877

7978
genotype = model.genotype()
80-
logger.info(f'Genotype: {genotype}\n')
79+
print(f'Genotype: {genotype}\n')
8180

8281
train(
8382
trainloader,
@@ -154,7 +153,7 @@ def train(trainloader,
154153
min_accuracy = accuracy_avg
155154

156155
if step % args.log_interval == 0:
157-
logger.info(f'Step: {step} loss: {meter.loss_meter.avg:.4}')
156+
print(f'Step: {step} loss: {meter.loss_meter.avg:.4}')
158157

159158
meter.update_epoch()
160159
meter.save(args.savepath)
@@ -179,7 +178,7 @@ def validate(validloader, model, criterion, args, tasks, meter, device):
179178
meter.update_batch_accuracy(prec1, batch_size)
180179

181180
if step % args.log_interval == 0:
182-
logger.info(f'>> Validation: {step} loss: {meter.loss_meter.avg:.4}')
181+
print(f'>> Validation: {step} loss: {meter.loss_meter.avg:.4}')
183182

184183
meter.update_epoch()
185184
meter.save(args.savepath)

0 commit comments

Comments
 (0)