Update loader.py
Browse files
loader.py
CHANGED
|
@@ -2,9 +2,6 @@ import torch
|
|
| 2 |
import numpy as np
|
| 3 |
from graph_decoder.diffusion_model import GraphDiT
|
| 4 |
|
| 5 |
-
# model_state = load_model()
|
| 6 |
-
# generate_graph(2.5, 15.4, 21.0, 1.5, 2.8, 2, 0, 1, model_state, 50)
|
| 7 |
-
|
| 8 |
def count_parameters(model):
|
| 9 |
r"""
|
| 10 |
Returns the number of trainable parameters and number of all parameters in the model.
|
|
|
|
| 2 |
import numpy as np
|
| 3 |
from graph_decoder.diffusion_model import GraphDiT
|
| 4 |
|
|
|
|
|
|
|
|
|
|
| 5 |
def count_parameters(model):
|
| 6 |
r"""
|
| 7 |
Returns the number of trainable parameters and number of all parameters in the model.
|