(images, targets, device)
| 16 | |
| 17 | |
| 18 | def to_device(images, targets, device): |
| 19 | images = [image.to(device) for image in images] |
| 20 | for t in targets: |
| 21 | t["boxes"] = t["boxes"].to(device) |
| 22 | t["labels"] = t["labels"].to(device) |
| 23 | return images, targets |
| 24 | |
| 25 | |
| 26 | def train_one_epoch(cfg, model, optimizer, data_loader, device, epoch, tfboard, softmax_criterion_s2, softmax_criterion_s3): |
no outgoing calls
no test coverage detected