update scripts-cluster

2019-03-31 22:49:43 +08:00
parent 280c9f3099
commit 4bac459bf9
20 changed files with 118 additions and 1248 deletions
--- a/lib/utils/model_utils.py
+++ b/lib/utils/model_utils.py
@@ -2,6 +2,7 @@ import torch
 import torch.nn as nn
 import numpy as np

+
 def count_parameters_in_MB(model):
  if isinstance(model, nn.Module):
    return np.sum(np.prod(v.size()) for v in model.parameters())/1e6