谁来救救我。。。运行时报错#EOFError: Ran out of input

[pk-tzy]

for epoch in range(cfg.starting_epoch, cfg.num_epochs):
    train_one_epoch_acc(
        model=model,
        optimizer=optimizer,
        data_loader=train_loader,
        epoch=epoch,
        print_freq=cfg.print_freq,
        max_grad_norm=cfg.max_norm,
        accelerator=accelerator,
    )
    lr_scheduler.step()

main.py脚本进入循环train_one_epoch_acc报错。
错误信息如下：
_pickle.PicklingError: Can't pickle <function at 0x0000026F6BAAF790>: attribute lookup on transforms.presets failed
self = reduction.pickle.load(from_parent)
EOFError: Ran out of input