We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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):
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
The text was updated successfully, but these errors were encountered:
Hi, could you be more specific about where and what you modified? What you pasted here is not easy to read.
Sorry, something went wrong.
No branches or pull requests
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)
return teacher_outputs
The text was updated successfully, but these errors were encountered: