text: #######################################################################
  # Copyright (C) 2017 Shangtong Zhang(zhangshangtong.cpp@gmail.com)    #
  # Permission given to modify the code as long as you keep this        #
  # declaration at the top                                              #
  #######################################################################
  
  import torch.nn as nn
  
  
  class BaseNet:
      def __init__(self):
          pass
  
  
  def layer_init(layer, w_scale=1.0):
      nn.init.orthogonal_(layer.weight.data)
      layer.weight.data.mul_(w_scale)
      nn.init.constant_(layer.bias.data, 0)
      return layer
  