rodeo icon indicating copy to clipboard operation
rodeo copied to clipboard

Where is get_data128 function?

Open 2698022795 opened this issue 1 year ago • 0 comments

So pleasure to view this project. I have a question about get_distillationinfo in train_better.py

def get_distillinfo(model, dl):
    save = {}
    print("dumping info ......")
    model.eval()
    with torch.no_grad():
        for ii, (images, targets) in tqdm(enumerate(dl), total=len(dl)):
            images = list(image.to(device) for image in images)
            targets = [{k: v.to(device) for k, v in t.items()} for t in targets]
            for image, target in zip(images, targets):
                image_id = '{0:06d}'.format(target['image_id'].item())
                info = model.get_data128([image], [target])
                save[image_id] = info
        return save

where can i find the get_data128() function? I don't see it in frcnn_mod.py ?

2698022795 avatar May 04 '23 16:05 2698022795