Dan Sukhorukov

@ongiaf Do you have any success decoding FuXi (I've been fine-tuning this model for a long time)? I recommend paying attention to [this solution](https://github.com/lizhuoq/WeatherLearn/blob/master/weatherlearn/models/fuxi/fuxi.py)

@ongiaf Did you manage to run FuXi with the current weights for the fine-tuning process (I am currently thinking about how to complete the work on the model on a...

@juanqiu1 In order for this to work with FuXi, you will need to change the [`onnx2torch/node_converters/layer_norm.py` ](https://github.com/ENOT-AutoDL/onnx2torch/blob/main/onnx2torch/node_converters/layer_norm.py#L58) parameter to [1536]: ```python @add_converter(operation_type='LayerNormalization', version=17) def _(node: OnnxNode, graph: OnnxGraph) -> OperationConverterResult:...

@juanqiu1 Yes, I managed to start the learning process by highlighting the `named_parameters()` part within the last dozen _UTransformer_ (this was enough for fine-tunning). I used Nvidia A100 (40GB).