config.json: 100%|██████████| 945/945 [00:00<00:00, 7.44MB/s]
pytorch_model.bin: 100%|██████████| 6.60M/6.60M [00:01<00:00, 6.35MB/s]
/home/raghu/anaconda3/envs/optim/lib/python3.11/site-packages/transformers/models/decision_transformer/modeling_decision_transformer.py:550: TracerWarning: Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs!
  if batch_size <= 0:
		-[x] shape (2, 16, 11) doesn't match (2, 16, 1)
		-[x] shape (2, 16, 11) doesn't match (2, 16, 1)
		-[x] shape (2, 16, 11) doesn't match (2, 16, 1)
		-[x] shape (2, 16, 11) doesn't match (2, 16, 1)
		-[x] values not close enough, max diff: 2.3578810691833496 (atol: 1e-05)
		-[x] values not close enough, max diff: 2.3578810691833496 (atol: 1e-05)
		-[x] values not close enough, max diff: 2.3578810691833496 (atol: 1e-05)
		-[x] values not close enough, max diff: 2.3578810691833496 (atol: 1e-05)
		-[x] shape (2, 16, 1) doesn't match (2, 16, 11)
		-[x] shape (2, 16, 1) doesn't match (2, 16, 11)
		-[x] shape (2, 16, 1) doesn't match (2, 16, 11)
		-[x] shape (2, 16, 1) doesn't match (2, 16, 11)
		-[x] values not close enough, max diff: 2.3578813076019287 (atol: 1e-05)
		-[x] values not close enough, max diff: 2.3578813076019287 (atol: 1e-05)
		-[x] values not close enough, max diff: 2.3578813076019287 (atol: 1e-05)
		-[x] values not close enough, max diff: 2.3578813076019287 (atol: 1e-05)
Traceback (most recent call last):

  File ~/anaconda3/envs/optim/lib/python3.11/site-packages/spyder_kernels/py3compat.py:356 in compat_exec
    exec(code, globals, locals)

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum-cli.py:17
    sys.exit(main())

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/commands/optimum_cli.py:208 in main
    service.run()

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/commands/export/onnx.py:265 in run
    main_export(

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/__main__.py:373 in main_export
    onnx_export_from_model(

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/convert.py:1256 in onnx_export_from_model
    raise e

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/convert.py:1243 in onnx_export_from_model
    validate_models_outputs(

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/convert.py:184 in validate_models_outputs
    raise exceptions[-1][1]

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/convert.py:167 in validate_models_outputs
    validate_model_outputs(

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/convert.py:237 in validate_model_outputs
    _run_validation(

  File ~/DL/topics/RL/Godot/104-Godot-RL/optimum/optimum/exporters/onnx/convert.py:427 in _run_validation
    raise ShapeError(f"Output shapes do not match between reference model and ONNX exported model:\n{msg}")

ShapeError: Output shapes do not match between reference model and ONNX exported model:
- return_preds: got (2, 16, 1) (reference) and (2, 16, 11) (ONNX)
- state_preds: got (2, 16, 11) (reference) and (2, 16, 1) (ONNX)