關鍵是:
costs = sorted(costs,key = lambda x:x[0] - x [1])
其作用是將
[[10,20],[30,200],[400,50],[30,20]],變成
[[30, 200], [10, 20], [30, 20], [400, 50]],這樣的話取前2個的第一個值和後兩個的第二個值就可以了
關鍵是:
costs = sorted(costs,key = lambda x:x[0] - x [1])
其作用是將
[[10,20],[30,200],[400,50],[30,20]],變成
[[30, 200], [10, 20], [30, 20], [400, 50]],這樣的話取前2個的第一個值和後兩個的第二個值就可以了
利用torchsummary觀察每一層的情況 1)按照方式 pip install torchsummary 2)