mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

replace self.model() with self.forward

pull/1116/head
weishengyu 2021-08-08 02:05:26 +08:00
parent 6c164c9e2b
commit 91af66dccd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/engine/trainer.py
View File

@ -597,7 +597,7 @@ class Trainer(object):
if len(batch) == 3: if len(batch) == 3:
has_unique_id = True has_unique_id = True
batch[2] = batch[2].reshape([-1, 1]).astype("int64") batch[2] = batch[2].reshape([-1, 1]).astype("int64")
out = self.model(batch[0], batch[1]) out = self.forward(batch)
batch_feas = out["features"] batch_feas = out["features"]
# do norm # do norm
@ -662,7 +662,7 @@ class Trainer(object):
image_file_list.append(image_file) image_file_list.append(image_file)
if len(batch_data) >= batch_size or idx == len(image_list) - 1: if len(batch_data) >= batch_size or idx == len(image_list) - 1:
batch_tensor = paddle.to_tensor(batch_data) batch_tensor = paddle.to_tensor(batch_data)
out = self.model(batch_tensor) out = self.forward(batch_tensor)
if isinstance(out, list): if isinstance(out, list):
out = out[0] out = out[0]
result = postprocess_func(out, image_file_list) result = postprocess_func(out, image_file_list)