|
@@ -103,8 +103,8 @@ class Trainer(BaseTrainer):
|
|
|
for imgs, targets in data_loader_train:
|
|
for imgs, targets in data_loader_train:
|
|
|
imgs = move_to_device(imgs, device)
|
|
imgs = move_to_device(imgs, device)
|
|
|
targets=move_to_device(targets,device)
|
|
targets=move_to_device(targets,device)
|
|
|
- print(f'imgs:{len(imgs)}')
|
|
|
|
|
- print(f'targets:{len(targets)}')
|
|
|
|
|
|
|
+ # print(f'imgs:{len(imgs)}')
|
|
|
|
|
+ # print(f'targets:{len(targets)}')
|
|
|
losses = model(imgs, targets)
|
|
losses = model(imgs, targets)
|
|
|
# print(losses)
|
|
# print(losses)
|
|
|
loss = _loss(losses)
|
|
loss = _loss(losses)
|
