Unexpected GPU space hike when using list to save predictions in pytorch

Question

num_epochs=1
all_x = []
all_y = []
all_y_hat = []
for epoch in tqdm(range(num_epochs)):
        #Evaluation and training on training dataset
       
        for i,(i