Fix missed part in commit ac073ab.

Author: ruotianluo

tools/convert_from_tensorflow.py

@@ -61,14 +61,14 @@
         var_dict[k.replace(m.group(0), str(int(m.group(1)) - 1))] = var_dict[k]
         del var_dict[k]
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if var_dict[k].ndim == 4:
         var_dict[k] = var_dict[k].transpose((3, 2, 0, 1)).copy(order='C')
     if var_dict[k].ndim == 2:
         var_dict[k] = var_dict[k].transpose((1, 0)).copy(order='C')
     # assert x[k].shape == var_dict[k].shape, k
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     var_dict[k] = torch.from_numpy(var_dict[k])
 
 

tools/convert_from_tensorflow_vgg.py

@@ -66,7 +66,7 @@
             var_dict[k.replace(a,b)] = var_dict[k]
             del var_dict[k]
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     if 'classifier.0' in k:
         if var_dict[k].ndim == 2: # weight
             var_dict[k] = var_dict[k].reshape(7,7,512,4096).transpose((3, 2, 0, 1)).reshape(4096, -1).copy(order='C')
@@ -77,7 +77,7 @@
             var_dict[k] = var_dict[k].transpose((1, 0)).copy(order='C')
     # assert x[k].shape == var_dict[k].shape, k
 
-for k in var_dict.keys():
+for k in list(var_dict.keys()):
     var_dict[k] = torch.from_numpy(var_dict[k])
 
 torch.save(var_dict, args.tensorflow_model[:args.tensorflow_model.find('.ckpt')]+'.pth')