Skip to content

Commit c2411f6

Browse files
committed
fix
1 parent 19ff5ad commit c2411f6

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

app/Converters/parser.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ def __init__(self, **kwargs):
1818

1919

2020
BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
21-
MODEL_PATH = os.path.join(BASE_DIR, 'docs\\models', 'AlexNet-model.h5')
22-
MODEL_DATA_PATH = os.path.join(BASE_DIR, 'docs\\jsons', 'model_data_alexnet_1.json')
21+
MODEL_PATH = os.path.join(BASE_DIR, 'docs/models', 'AlexNet-model.h5')
22+
MODEL_DATA_PATH = os.path.join(BASE_DIR, 'docs/jsons', 'model_data_alexnet_1.json')
2323

2424
model = load_model(MODEL_PATH, custom_objects={'GlorotUniform': CustomGlorotUniform, 'Zeros': CustomZeros})
2525

include/Weights_Reader/reader_weights.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,4 +10,4 @@ using namespace itlab_2023;
1010
json read_json(const std::string& filename);
1111
void extract_values_from_json(const json& j, std::vector<float>& values);
1212
void parse_json_shape(const json& j, std::vector<size_t>& shape, size_t dim);
13-
Tensor create_tensor_from_json(const json& j, Type type);
13+
Tensor create_tensor_from_json(const json& layer_data, Type type);

0 commit comments

Comments
 (0)