-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsave_t7_to_mat.lua
executable file
·86 lines (70 loc) · 3.26 KB
/
save_t7_to_mat.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
require 'optim'
cmd = torch.CmdLine()
cmd:option('-expnum',0,'experiment number')
opt = cmd:parse(arg)
print('Experiment number = '..opt.expnum)
local nOutput = 42
load_dir = 'exp/'..opt.expnum
loss_tr_epoch = torch.load(load_dir..'/loss_tr_epoch')
loss_te_epoch = torch.load(load_dir..'/loss_te_epoch')
loss_val_epoch = torch.load(load_dir..'/loss_val_epoch')
acc_tr_epoch = torch.load(load_dir..'/acc_tr_epoch')
acc_te_epoch = torch.load(load_dir..'/acc_te_epoch')
acc_val_epoch = torch.load(load_dir..'/acc_val_epoch')
--[[
confusion_tr = torch.load(load_dir..'/confusion_tr')
confusion_te = torch.load(load_dir..'/confusion_te')
--]]
loss_each_tr = torch.load(load_dir..'/loss_each_tr')
loss_each_te = torch.load(load_dir..'/loss_each_te')
loss_each_val = torch.load(load_dir..'/loss_each_val')
correct_each_tr = torch.load(load_dir..'/correct_each_tr')
correct_each_te = torch.load(load_dir..'/correct_each_te')
correct_each_val = torch.load(load_dir..'/correct_each_val')
confusion_tr_table = torch.load(load_dir..'/confusion_tr')
confusion_te_table = torch.load(load_dir..'/confusion_te')
confusion_val_table = torch.load(load_dir..'/confusion_val')
confusion_tr = torch.Tensor(nOutput,nOutput)
confusion_te = torch.Tensor(nOutput,nOutput)
confusion_val = torch.Tensor(nOutput,nOutput)
for i=1,nOutput do
for j=1,nOutput do
confusion_tr[i][j] = confusion_tr_table.mat[i][j]
confusion_te[i][j] = confusion_te_table.mat[i][j]
confusion_val[i][j] = confusion_val_table.mat[i][j]
end
end
-- mattorch = require 'mattorch'\
matio = require 'matio'
matio.save(load_dir..'/loss_tr_epoch.mat',loss_tr_epoch)
matio.save(load_dir..'/loss_te_epoch.mat',loss_te_epoch)
matio.save(load_dir..'/loss_val_epoch.mat',loss_val_epoch)
matio.save(load_dir..'/acc_tr_epoch.mat',acc_tr_epoch)
matio.save(load_dir..'/acc_te_epoch.mat',acc_te_epoch)
matio.save(load_dir..'/acc_val_epoch.mat',acc_val_epoch)
matio.save(load_dir..'/loss_each_tr.mat',loss_each_tr)
matio.save(load_dir..'/loss_each_te.mat',loss_each_te)
matio.save(load_dir..'/loss_each_val.mat',loss_each_val)
matio.save(load_dir..'/correct_each_tr.mat',correct_each_tr)
matio.save(load_dir..'/correct_each_te.mat',correct_each_te)
matio.save(load_dir..'/correct_each_val.mat',correct_each_val)
matio.save(load_dir..'/confusion_tr.mat',confusion_tr)
matio.save(load_dir..'/confusion_te.mat',confusion_te)
matio.save(load_dir..'/confusion_val.mat',confusion_val)
--[[
mattorch.save(load_dir..'/loss_tr_epoch.mat',loss_tr_epoch)
mattorch.save(load_dir..'/loss_te_epoch.mat',loss_te_epoch)
mattorch.save(load_dir..'/loss_val_epoch.mat',loss_val_epoch)
mattorch.save(load_dir..'/acc_tr_epoch.mat',acc_tr_epoch)
mattorch.save(load_dir..'/acc_te_epoch.mat',acc_te_epoch)
mattorch.save(load_dir..'/acc_val_epoch.mat',acc_val_epoch)
mattorch.save(load_dir..'/loss_each_tr.mat',loss_each_tr)
mattorch.save(load_dir..'/loss_each_te.mat',loss_each_te)
mattorch.save(load_dir..'/loss_each_val.mat',loss_each_val)
mattorch.save(load_dir..'/correct_each_tr.mat',correct_each_tr)
mattorch.save(load_dir..'/correct_each_te.mat',correct_each_te)
mattorch.save(load_dir..'/correct_each_val.mat',correct_each_val)
mattorch.save(load_dir..'/confusion_tr.mat',confusion_tr)
mattorch.save(load_dir..'/confusion_te.mat',confusion_te)
mattorch.save(load_dir..'/confusion_val.mat',confusion_val)
--]]