参数不是具有与设备相同长度的列表

Last few logs: Defaulting to GELU activation (see here: https://arxiv.org/abs/1606.08415) Defaulting to GELU activation (see here: https://arxiv.org/abs/1606.08415) Defaulting to GELU activation (see here: https://arxiv.org/abs/1606.08415) prediction_loop marked as finished Raising captured error Traceback (most recent call last): File "main.py", line 257, in <module> main(args) File "main.py", line 184, in main handle_pred_output_fn(predictions, logger, enc, params, out_name=f"predictions_{args.sacred_id}_{current_step}") File "C:\Max\GPTNeo\inputs.py", line 165, in handle_pred_output for i, p in enumerate(predictions): File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\tpu_estimator.py", line 3153, in predict rendezvous.raise_errors() File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\error_handling.py", line 150, in raise_errors six.reraise(typ, value, traceback) File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\six.py", line 703, in reraise raise value File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\tpu_estimator.py", line 3142, in predict for result in super(TPUEstimator, self).predict( File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\estimator.py", line 612, in predict estimator_spec = self._call_model_fn(features, None, ModeKeys.PREDICT, File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\tpu_estimator.py", line 2941, in _call_model_fn return super(TPUEstimator, self)._call_model_fn(features, labels, mode, File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\estimator.py", line 1163, in _call_model_fn model_fn_results = self._model_fn(features=features, **kwargs) File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\tpu_estimator.py", line 3199, in _model_fn estimator_spec = model_fn_wrapper.call_without_tpu( File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\tpu_estimator.py", line 1729, in call_without_tpu return self._call_model_fn(features, labels, is_export_mode=is_export_mode) File "C:\Users\jerkm\AppData\Local\Programs\Python\Python38\lib\site-packages\tensorflow_estimator\Python\estimator\tpu\tpu_estimator.py", line 2072, in _call_model_fn estimator_spec = self._model_fn(features=features, **kwargs) File "C:\Max\GPTNeo\model_fns.py", line 112, in model_fn lowering = mtf.Lowering(graph, {mesh: mesh_impl}, autostack=True) File