From a65ec5c69373fc0b85348f85ff897a662afe8580 Mon Sep 17 00:00:00 2001 From: Rudeus Date: Wed, 8 Oct 2025 18:43:26 +0530 Subject: [PATCH] fix fromarray depreceation (#12512) --- examples/train_resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/train_resnet.py b/examples/train_resnet.py index 8feee80820..d15e05e450 100755 --- a/examples/train_resnet.py +++ b/examples/train_resnet.py @@ -32,7 +32,7 @@ if __name__ == "__main__": lr = 5e-3 transform = ComposeTransforms([ - lambda x: [Image.fromarray(xx, mode='L').resize((64, 64)) for xx in x], + lambda x: [Image.fromarray(xx).resize((64, 64)) for xx in x], lambda x: np.stack([np.asarray(xx) for xx in x], 0), lambda x: x / 255.0, lambda x: np.tile(np.expand_dims(x, 1), (1, 3, 1, 1)).astype(np.float32),