mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-06 03:27:44 +00:00
Xinjie Shen
f20dc4482b
* upload example code for model implementation * Refactor Model Implement * add export --------- Co-authored-by: Young <afe.young@gmail.com>
19 lines
1.1 KiB
YAML
19 lines
1.1 KiB
YAML
|
|
|
|
code_implement_sys: -|
|
|
You are an assistant whose job is to answer user's question."
|
|
code_implement_user: -|
|
|
With the following given information, write a python code using pytorch and torch_geometric to implement the model.
|
|
This model is in the graph learning field, only have one layer.
|
|
The input will be node_feature [num_nodes, dim_feature] and edge_index [2, num_edges] (It would be the input of the forward model)
|
|
There is not edge attribute or edge weight as input. The model should detect the node_feature and edge_index shape, if there is Linear transformation layer in the model, the input and output shape should be consistent. The in_channels is the dimension of the node features.
|
|
Implement the model forward function based on the following information:model formula information.
|
|
1. model name:{{name}}
|
|
2. model description:{{description}}
|
|
3. model formulation:{{formulation}}
|
|
4. model variables:{{variables}}.
|
|
You must complete the forward function as far as you can do.
|
|
\# Execution
|
|
Your implemented code will be exectued in the follow way
|
|
{{execute_desc}}
|