2026年人工智能工程师职业技能提升试题及答案.docxVIP

2026年人工智能工程师职业技能提升试题及答案

一、单项选择题（每题2分，共20分）

1.在PyTorch2.3中，若需将模型权重以16-bit浮点形式保存且保留训练状态，下列做法最合理的是

A.`torch.save(model.state_dict(),ckpt.pt,dtype=torch.float16)`

B.`torch.save({model:model.half().state_dict(),opt:optimizer.state_dict()},ckpt.pt)`

C.`torch.save(model,ckpt.pt)`后手动`model.half()`

D.使用`torch.cuda.amp.state_dict()`保存

答案：B

解析：A项`torch.save`无`dtype`参数；C项保存完整模型易引入版本冲突；D项仅保存自动混合精度状态。B项显式将权重转半精度并同时保存优化器状态，兼顾体积与恢复需求。

2.对于StableDiffusionXL的UNet，若显存不足需降低激活值内存，下列方案中不改变推理结果的是

A.将`use_checkpoint=True`

B.使用`torch.backends.cudnn.benchmark=False`

C.替换`nn.GELU`为`nn.ReLU`

D.将`attention_