Python get_default_network Examples

Example #1

File: gaussian_vae.py Project: jxzhangjhu/NuX

  def get_generator_network(self):

    # The user can specify a custom network
    if self.create_network is not None:
      return self.create_network(self.out_shape)

    return util.get_default_network(self.out_shape, network_kwargs=self.network_kwargs)

Example #2

File: gaussian_vae.py Project: jxzhangjhu/NuX

  def get_inference(self, out_shape):
    out_shape = out_shape[:-1] + (2*out_shape[-1],)

    # The user can specify a custom network
    if self.create_network is not None:
      return self.create_inference(out_shape)

    return util.get_default_network(out_shape, network_kwargs=self.inference_network_kwargs)

Example #3

File: residual.py Project: jxzhangjhu/NuX

    def get_network(self, out_shape):

        # The user can specify a custom network
        if self.create_network is not None:
            return self.create_network(out_shape)

        return util.get_default_network(out_shape,
                                        network_kwargs=self.network_kwargs,
                                        lipschitz=True)

Example #4

File: coupling_base.py Project: Information-Fusion-Lab-Umass/NuX

    def get_network(self, out_shape):
        if hasattr(self, "_network"):
            return self._network

        # The user can specify a custom network
        if self.create_network is not None:
            self._network = self.create_network(out_shape)

        else:
            self._network = util.get_default_network(
                out_shape, network_kwargs=self.network_kwargs)

        return self._network

Example #5

 def get_network(self, out_shape):
     if self.create_network is not None:
         return self.create_network(out_shape)
     return util.get_default_network(out_shape,
                                     network_kwargs=self.network_kwargs,
                                     lipschitz=True)