mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-04 02:37:44 +00:00
Xu Yang
6b626eb56d
* Commit init framework * Co-authored-by: Yuante Li (FESCO Adecco Human Resources) <v-yuanteli@microsoft.com> Co-authored-by: XianBW <XianBW@users.noreply.github.com> * add an import * refine the whole framework * benchmark related framework * fix black and isort errors * move requirements to folder * fix black again --------- Co-authored-by: Young <afe.young@gmail.com> Co-authored-by: xuyang1 <xuyang1@microsoft.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}}
|