训练模型问题

请问一下从小样本换成全部样本训练时，需要注意input_data.py和train_CNN_CTU64.py中哪里的修改，只修改了训练样本数量可以吗，训练报错在images=data [:, 0:4096].astype(np.float32),MemoryError: Unabale to allocate 1.22Gib for an array with shape (80000,4096)and data type float32