/home/raghu/anaconda3/envs/rl/bin/optimum-cli Automatic task detection to reinforcement-learning. Automatic task detection to reinforcement-learning. Automatic task detection to reinforcement-learning. Using the export variant default. Available variants are: - default: The default ONNX variant. Using the export variant default. Available variants are: - default: The default ONNX variant. Using the export variant default. Available variants are: - default: The default ONNX variant. ***** Exporting submodel 1/1: DecisionTransformerModel ***** ***** Exporting submodel 1/1: DecisionTransformerModel ***** ***** Exporting submodel 1/1: DecisionTransformerModel ***** Using framework PyTorch: 2.4.0 Using framework PyTorch: 2.4.0 Using framework PyTorch: 2.4.0 Overriding 1 configuration item(s) Overriding 1 configuration item(s) Overriding 1 configuration item(s) - use_cache -> False - use_cache -> False - use_cache -> False Post-processing the exported models... Post-processing the exported models... Post-processing the exported models... Deduplicating shared (tied) weights... Deduplicating shared (tied) weights... Deduplicating shared (tied) weights... Validating ONNX model /home/raghu/DL/topics/RL/Godot/onnx/model.onnx... Validating ONNX model /home/raghu/DL/topics/RL/Godot/onnx/model.onnx... Validating ONNX model /home/raghu/DL/topics/RL/Godot/onnx/model.onnx... -[✓] ONNX model output names match reference model (return_preds, state_preds, action_preds, last_hidden_state) -[✓] ONNX model output names match reference model (return_preds, state_preds, action_preds, last_hidden_state) -[✓] ONNX model output names match reference model (return_preds, state_preds, action_preds, last_hidden_state) - Validating ONNX Model output "state_preds": - Validating ONNX Model output "state_preds": - Validating ONNX Model output "state_preds": -[✓] (2, 11, 11) matches (2, 11, 11) -[✓] (2, 11, 11) matches (2, 11, 11) -[✓] (2, 11, 11) matches (2, 11, 11) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) - Validating ONNX Model output "action_preds": - Validating ONNX Model output "action_preds": - Validating ONNX Model output "action_preds": -[✓] (2, 11, 3) matches (2, 11, 3) -[✓] (2, 11, 3) matches (2, 11, 3) -[✓] (2, 11, 3) matches (2, 11, 3) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) - Validating ONNX Model output "return_preds": - Validating ONNX Model output "return_preds": - Validating ONNX Model output "return_preds": -[✓] (2, 11, 1) matches (2, 11, 1) -[✓] (2, 11, 1) matches (2, 11, 1) -[✓] (2, 11, 1) matches (2, 11, 1) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) - Validating ONNX Model output "last_hidden_state": - Validating ONNX Model output "last_hidden_state": - Validating ONNX Model output "last_hidden_state": -[✓] (2, 33, 128) matches (2, 33, 128) -[✓] (2, 33, 128) matches (2, 33, 128) -[✓] (2, 33, 128) matches (2, 33, 128) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) -[x] values not close enough, max diff: nan (atol: 1e-05) The ONNX export succeeded with the warning: The maximum absolute difference between the output of the reference model and the ONNX exported model is not within the set tolerance 1e-05: - state_preds: max diff = nan - action_preds: max diff = nan - return_preds: max diff = nan - last_hidden_state: max diff = nan. The exported model was saved at: /home/raghu/DL/topics/RL/Godot/onnx The ONNX export succeeded with the warning: The maximum absolute difference between the output of the reference model and the ONNX exported model is not within the set tolerance 1e-05: - state_preds: max diff = nan - action_preds: max diff = nan - return_preds: max diff = nan - last_hidden_state: max diff = nan. The exported model was saved at: /home/raghu/DL/topics/RL/Godot/onnx The ONNX export succeeded with the warning: The maximum absolute difference between the output of the reference model and the ONNX exported model is not within the set tolerance 1e-05: - state_preds: max diff = nan - action_preds: max diff = nan - return_preds: max diff = nan - last_hidden_state: max diff = nan. The exported model was saved at: /home/raghu/DL/topics/RL/Godot/onnx state_preds ref_value: [[[ 0.331516 -0.21343133 0.13110688 -0.49706244 -0.34220165 0.5707443 -0.32417956 -1.125686 -0.17060848 -0.8151883 0.110229 ] [ 0.01389012 0.2697852 0.4046767 -0.80314994 -0.28623638 0.3052051 -0.01067244 -0.96100354 0.0883255 -0.7211001 0.06440777] [ 0.331516 -0.21343133 0.13110688 -0.49706244 -0.34220165 0.5707443 -0.32417956 -1.125686 -0.17060848 -0.8151883 0.110229 ] [-0.00844274 0.3924446 -0.03670885 -0.6666957 -0.29046276 0.46895903 0.19517688 -0.93204254 0.15587448 -0.8173044 0.12727764] [ 0.331516 -0.21343133 0.13110688 -0.49706244 -0.34220165 0.5707443 -0.32417956 -1.125686 -0.17060848 -0.8151883 0.110229 ] [-0.00844274 0.3924446 -0.03670885 -0.6666957 -0.29046276 0.46895903 0.19517688 -0.93204254 0.15587448 -0.8173044 0.12727764] [ 0.331516 -0.21343133 0.13110688 -0.49706244 -0.34220165 0.5707443 -0.32417956 -1.125686 -0.17060848 -0.8151883 0.110229 ] [-0.00844274 0.3924446 -0.03670885 -0.6666957 -0.29046276 0.46895903 0.19517688 -0.93204254 0.15587448 -0.8173044 0.12727764] [ 0.331516 -0.21343133 0.13110688 -0.49706244 -0.34220165 0.5707443 -0.32417956 -1.125686 -0.17060848 -0.8151883 0.110229 ] [-0.00844274 0.3924446 -0.03670885 -0.6666957 -0.29046276 0.46895903 0.19517688 -0.93204254 0.15587448 -0.8173044 0.12727764] [ 0.331516 -0.21343133 0.13110688 -0.49706244 -0.34220165 0.5707443 -0.32417956 -1.125686 -0.17060848 -0.8151883 0.110229 ]] [[ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan]]] state_preds ort_value: [[[ 0.33151606 -0.21343148 0.13110688 -0.49706215 -0.34220156 0.5707442 -0.3241794 -1.1256859 -0.17060864 -0.81518847 0.11022899] [ 0.01389008 0.26978517 0.40467688 -0.8031498 -0.28623626 0.30520505 -0.01067244 -0.9610035 0.0883256 -0.7211 0.06440772] [ 0.33151606 -0.21343148 0.13110688 -0.49706215 -0.34220156 0.5707442 -0.3241794 -1.1256859 -0.17060864 -0.81518847 0.11022899] [-0.00844273 0.39244464 -0.03670888 -0.6666955 -0.2904627 0.46895915 0.19517672 -0.9320425 0.1558742 -0.8173044 0.12727775] [ 0.33151606 -0.21343148 0.13110688 -0.49706215 -0.34220156 0.5707442 -0.3241794 -1.1256859 -0.17060864 -0.81518847 0.11022899] [-0.00844273 0.39244464 -0.03670888 -0.6666955 -0.2904627 0.46895915 0.19517672 -0.9320425 0.1558742 -0.8173044 0.12727775] [ 0.33151606 -0.21343148 0.13110688 -0.49706215 -0.34220156 0.5707442 -0.3241794 -1.1256859 -0.17060864 -0.81518847 0.11022899] [-0.00844273 0.39244464 -0.03670888 -0.6666955 -0.2904627 0.46895915 0.19517672 -0.9320425 0.1558742 -0.8173044 0.12727775] [ 0.33151606 -0.21343148 0.13110688 -0.49706215 -0.34220156 0.5707442 -0.3241794 -1.1256859 -0.17060864 -0.81518847 0.11022899] [-0.00844273 0.39244464 -0.03670888 -0.6666955 -0.2904627 0.46895915 0.19517672 -0.9320425 0.1558742 -0.8173044 0.12727775] [ 0.33151606 -0.21343148 0.13110688 -0.49706215 -0.34220156 0.5707442 -0.3241794 -1.1256859 -0.17060864 -0.81518847 0.11022899]] [[ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan] [ nan nan nan nan nan nan nan nan nan nan nan]]] action_preds ref_value: [[[-0.06170137 0.27931184 -0.16703954] [-0.32455266 -0.00717782 0.3923729 ] [-0.06170137 0.27931184 -0.16703954] [-0.42561153 -0.08041311 0.344285 ] [-0.15374242 0.12030884 -0.16778158] [-0.42561153 -0.08041311 0.344285 ] [-0.15374242 0.12030884 -0.16778158] [-0.42561153 -0.08041311 0.344285 ] [-0.15374242 0.12030884 -0.16778158] [-0.42561153 -0.08041311 0.344285 ] [-0.15374242 0.12030884 -0.16778158]] [[ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan]]] action_preds ort_value: [[[-0.06170133 0.27931187 -0.16703959] [-0.32455257 -0.00717806 0.39237314] [-0.06170133 0.27931187 -0.16703959] [-0.42561147 -0.08041307 0.34428483] [-0.15374258 0.12030909 -0.1677815 ] [-0.42561147 -0.08041307 0.34428483] [-0.15374258 0.12030909 -0.1677815 ] [-0.42561147 -0.08041307 0.34428483] [-0.15374258 0.12030909 -0.1677815 ] [-0.42561147 -0.08041307 0.34428483] [-0.15374258 0.12030909 -0.1677815 ]] [[ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan] [ nan nan nan]]] return_preds ref_value: [[[ 0.03254916] [-0.2789334 ] [ 0.03254916] [-0.10907605] [ 0.03254916] [-0.10907605] [ 0.03254916] [-0.10907605] [ 0.03254916] [-0.10907605] [ 0.03254916]] [[ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan]]] return_preds ort_value: [[[ 0.03254895] [-0.27893335] [ 0.03254895] [-0.10907608] [ 0.03254895] [-0.10907608] [ 0.03254895] [-0.10907608] [ 0.03254895] [-0.10907608] [ 0.03254895]] [[ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan] [ nan]]] last_hidden_state ref_value: [[[-0.15844241 -0.5158109 0.29880452 ... -0.11182172 0.48948517 -1.4883826 ] [ 0.04315017 -0.42534652 0.12423295 ... -0.74502057 1.2958332 -0.42410728] [ 1.178071 0.15747224 -0.4127394 ... -1.3206289 0.87288594 -1.2575938 ] ... [ 0.24832314 -0.4248045 0.39947343 ... -0.6062763 0.7227892 -1.3896236 ] [ 0.17837885 -0.4434575 0.11333983 ... -0.88618267 1.2839592 -0.4472614 ] [ 1.178071 0.15747224 -0.4127394 ... -1.3206289 0.87288594 -1.2575938 ]] [[ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan] ... [ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan]]] last_hidden_state ort_value: [[[-0.15844256 -0.51581085 0.29880428 ... -0.1118217 0.48948514 -1.4883827 ] [ 0.04315007 -0.4253464 0.12423277 ... -0.7450206 1.2958333 -0.4241073 ] [ 1.1780713 0.15747228 -0.41273937 ... -1.3206289 0.87288606 -1.2575942 ] ... [ 0.24832292 -0.42480445 0.39947337 ... -0.6062762 0.72278917 -1.3896238 ] [ 0.17837858 -0.44345766 0.11333981 ... -0.8861822 1.2839595 -0.44726127] [ 1.1780713 0.15747228 -0.41273937 ... -1.3206289 0.87288606 -1.2575942 ]] [[ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan] ... [ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan] [ nan nan nan ... nan nan nan]]]