rodeo
rodeo copied to clipboard
Where is get_data128 function?
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
?