'Tensor' object is not callable
youyeg opened this issue · 1 comments
I modified the code, and I get an error, does anybody have any idea why?
I am using CPU:
I have an error in this line:
—> 10 output_teacher_batch = teacher_model(data_batch).data().numpy()
TypeError: ‘Tensor’ object is not callable
Does anybody have an idea how to solve this?
def fetch_teacher_outputs(teacher_model, dataloader):
set teacher_model to evaluation mode
teacher_model.eval()
teacher_outputs = []
for i, (data_batch, labels_batch) in enumerate(dataloader):
if torch.cuda.is_available():
data_batch, labels_batch = data_batch.cuda(async=True),
labels_batch.cuda(async=True)
data_batch, labels_batch = Variable(data_batch), Variable(labels_batch)
**output_teacher_batch = teacher_model(data_batch).data().numpy()**
teacher_outputs.append(output_teacher_batch)
return teacher_outputs
Hi, could you be more specific about where and what you modified? What you pasted here is not easy to read.