Does your model accept multiple inputs?
If images are separate inputs, do:
inputs = [x.to(device) for x in images]
outputs = AE(*inputs)

if images are single inputs, as a batch, use the collate_fn to process the list of samples to form a batch.
Roy