kevinmevin commited on
Commit
2396665
·
1 Parent(s): a455dcb

Add coefficients for model

Browse files
model_coefficients/README.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Model coeficients
2
+ Here in this folder you can find the last and best coeficients obtained during training for each stage. To load these coeficients in the following way
3
+ ```python
4
+ >>> import MSECNN
5
+ >>> import train_model_utils
6
+ >>>
7
+ >>> # Initialize parameters
8
+ >>> path_to_folder_with_model_params = "model_coefficients/best_coefficients"
9
+ >>> device = "cuda:0"
10
+ >>> qp = 32 # Quantisation Parameter
11
+ >>>
12
+ >>> # Initialize Model
13
+ >>> stg1_2 = msecnn.MseCnnStg1(device=device, QP=qp).to(device)
14
+ >>> stg3 = msecnn.MseCnnStgX(device=device, QP=qp).to(device)
15
+ >>> stg4 = msecnn.MseCnnStgX(device=device, QP=qp).to(device)
16
+ >>> stg5 = msecnn.MseCnnStgX(device=device, QP=qp).to(device)
17
+ >>> stg6 = msecnn.MseCnnStgX(device=device, QP=qp).to(device)
18
+ >>> model = (stg1_2, stg3, stg4, stg5, stg6)
19
+ >>>
20
+ >>> model = train_model_utils.load_model_parameters_eval(model, path_to_folder_with_model_params, device) # Load for all of the stages
21
+ ```
model_coefficients/best_coefficients/stg_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71561b982b84ae348fc06daf7c8558d4c56523d073058ddbbb0f81b8a528f3ad
3
+ size 104435
model_coefficients/best_coefficients/stg_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:560d1e5c6df6ae31097a5539d86d8a10894c077f72a7f2a89106ebf74a03decd
3
+ size 752885
model_coefficients/best_coefficients/stg_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:269da8310b17320985ac4baac2254f4ec8e7e37d1caf35010cdabccf712a2f2e
3
+ size 752885
model_coefficients/best_coefficients/stg_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50fc1ef6890dc99c9a7b98fc4496053c255cc6908184b6a6edf04b8d053189d7
3
+ size 752885
model_coefficients/best_coefficients/stg_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65a1af7bdce0a4d7edbbd33f0e90cca75459f70fbedb56f40115f7de4e0c3bad
3
+ size 752885
model_coefficients/last_coefficients/stg_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71561b982b84ae348fc06daf7c8558d4c56523d073058ddbbb0f81b8a528f3ad
3
+ size 104435
model_coefficients/last_coefficients/stg_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:560d1e5c6df6ae31097a5539d86d8a10894c077f72a7f2a89106ebf74a03decd
3
+ size 752885
model_coefficients/last_coefficients/stg_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:269da8310b17320985ac4baac2254f4ec8e7e37d1caf35010cdabccf712a2f2e
3
+ size 752885
model_coefficients/last_coefficients/stg_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50fc1ef6890dc99c9a7b98fc4496053c255cc6908184b6a6edf04b8d053189d7
3
+ size 752885
model_coefficients/last_coefficients/stg_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d926fd057d9e93b134f68748c5fbb05028d4479667f0138aaaf03bd2e29f3d0b
3
+ size 752885