MCPcopy Create free account

hub / github.com/YorkUCVIL/Wavelet-Flow / functions

Functions227 in github.com/YorkUCVIL/Wavelet-Flow

↓ 52 callersFunctiontlog
prints a string with an appropriate header
src/util/tlog.py:3
↓ 14 callersFunctionto_attributes
convert dictionary to object with keys as attributes should implement with a class inheriting dict but would cause leak in 2.7 ignores non dict ob
src/util/to_attributes.py:2
↓ 7 callersFunctionget_checkpoint_path
(path, iter)
src/util/check_resume.py:38
↓ 6 callersMethodforward
(self,x,init=False,conditioning=None)
src/tf_components/bijector/Step.py:27
↓ 6 callersFunctionget_latest_checkpoint_iter
(checkpoint_path,checkpoint_dir_contents)
src/util/check_resume.py:14
↓ 5 callersFunctionbpd_metric
bits per dimension metric
src/tf_components/bpd_metric.py:4
↓ 5 callersFunctionget_checkpoints
(path)
src/util/check_resume.py:30
↓ 5 callersFunctionload_config
(instance_dir,relative_path)
src/util/load_config.py:8
↓ 5 callersFunctionsession_setup
container for tensorflow config
src/util/session_setup.py:3
↓ 4 callersMethoddata_to_latent_and_log_density
forward pass, must implement
src/models/shared/Normalizing_flow.py:9
↓ 4 callersMethodget_variables
implement for shared method to collect variables
src/tf_components/Layer.py:28
↓ 4 callersFunctioninit_random_seeds
()
src/util/init_random_seeds.py:6
↓ 3 callersFunctionask_user_bool
(prompt)
src/util/check_resume.py:62
↓ 3 callersMethodinverse
(self,y,conditioning=None)
src/tf_components/bijector/Step.py:47
↓ 3 callersMethodlatent_to_super_res
(self,latent,level,base)
src/models/shared/Multi_scale_flow.py:129
↓ 3 callersMethodsample_latents
samples all latents required to generate data(s)
src/models/shared/Normalizing_flow.py:60
↓ 3 callersFunctionsave_grid_image
(grid, path, scale=4)
src/util/save_grid_image.py:4
↓ 3 callersFunctionuniform_dequant_bound
converts a continuous log density to a lower bound on the log probability in respect to uniform dequant, assumes values are compressed within [0,1]
src/tf_components/uniform_dequant_bound.py:3
↓ 3 callersMethodupdate
(self,up_dict)
src/util/to_attributes.py:11
↓ 2 callersFunctionclear_checkpoints
(checkpoint_path,checkpoint_dir_contents)
src/util/check_resume.py:9
↓ 2 callersFunctionclear_logs
(log_path,log_dir_contents)
src/util/check_resume.py:4
↓ 2 callersMethodconcat_features
(self,x1,x2)
src/tf_components/bijector/Coupling.py:158
↓ 2 callersFunctionedge_bias
injects a mask into a tensor for 2d convs to indicate padding locations
src/tf_components/edge_bias.py:3
↓ 2 callersMethodget_haar_kernel
generates kernel for haar wavelet downsampling kernel should be inverse of itself
src/tf_components/bijector/Haar_squeeze.py:17
↓ 2 callersMethodget_kernel
(self,x)
src/tf_components/bijector/Invertible_1x1.py:21
↓ 2 callersMethodget_kernel
(self,x)
src/tf_components/bijector/Invertible_1x1.py:107
↓ 2 callersMethodget_mean_log_var
(self, shape)
src/tf_components/bijector/Element_shift_scale.py:16
↓ 2 callersMethodget_mean_log_var
(self, channels)
src/tf_components/bijector/Act_norm.py:16
↓ 2 callersMethodget_variables
(self, filter=None)
src/tf_components/bijector/Coupling.py:168
↓ 2 callersMethodlatent_log_density
latent variable is diagonal unit gaussian
src/models/shared/Normalizing_flow.py:83
↓ 2 callersMethodlatent_to_data
inverse pass, must implement
src/models/shared/Normalizing_flow.py:18
↓ 2 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/imagenet_32_haar/Training_data.py:63
↓ 2 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/ffhq_1024_haar/Training_data.py:64
↓ 2 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/celeba_1024_haar/Training_data.py:64
↓ 2 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/lsun_bedroom_64_haar/Training_data.py:67
↓ 2 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/imagenet_64_haar/Training_data.py:63
↓ 2 callersMethodsplit_features
(self,x)
src/tf_components/bijector/Coupling.py:144
↓ 2 callersMethodst
(self,x,out_size,init=False,conditioning=None)
src/tf_components/bijector/Coupling.py:57
↓ 1 callersFunctionInvertible_1x1
(use_lu=False,*args,**kwargs)
src/tf_components/bijector/Invertible_1x1.py:7
↓ 1 callersMethod__init__
(self,dtype = tf.float32, collection=None, name='invertible_1x1_lu')
src/tf_components/bijector/Invertible_1x1.py:15
↓ 1 callersMethod__init__
(self, k, width, edge_bias, conditional=False, collection=None, name='coupling')
src/tf_components/bijector/Coupling.py:32
↓ 1 callersMethodadapt_conditioning
(self,conditioning)
src/tf_components/bijector/Coupling.py:137
↓ 1 callersFunctionbuild_training_graph_partial
(config,partial_level)
src/models/ffhq_1024_haar/build_training_graph.py:6
↓ 1 callersFunctioncheck_resume
detects/manages files created during previous runs assumes log file is always created if checkpoints exist future: add option to always train fres
src/util/check_resume.py:43
↓ 1 callersMethodcompute_latent_shapes
computes shapes of latents given data shape assumes we split the latent in half every squeeze except last
src/models/shared/Normalizing_flow.py:31
↓ 1 callersMethodforward
implement forward pass
src/tf_components/bijector/Bijector.py:14
↓ 1 callersMethodforward_conv
(self,x)
src/tf_components/bijector/Haar_squeeze.py:107
↓ 1 callersMethodget_conv_haar_kernel
(self,n_channels)
src/tf_components/bijector/Haar_squeeze.py:51
↓ 1 callersFunctionget_logs
(path)
src/util/check_resume.py:21
↓ 1 callersMethodget_variables
(self,filter=None)
src/models/imagenet_32_haar/Conditioning_network.py:41
↓ 1 callersMethodget_variables
get variables for training, must implement
src/models/shared/Normalizing_flow.py:24
↓ 1 callersMethodget_variables
(self,filter=None)
src/models/ffhq_1024_haar/Conditioning_network.py:61
↓ 1 callersMethodget_variables
(self,filter=None)
src/models/celeba_1024_haar/Conditioning_network.py:61
↓ 1 callersMethodget_variables
(self,filter=None)
src/models/lsun_bedroom_64_haar/Conditioning_network.py:45
↓ 1 callersMethodget_variables
(self,filter=None)
src/models/imagenet_64_haar/Conditioning_network.py:45
↓ 1 callersMethodinverse
(self,base,detail)
src/tf_components/bijector/Haar_squeeze_split.py:46
↓ 1 callersFunctioniteration_print
(header, values)
src/util/iteration_print.py:4
↓ 1 callersMethodlatent_to_data
inverse pass
src/models/shared/Single_scale_flow.py:65
↓ 1 callersMethodlatent_to_data
inverse pass
src/models/shared/Multi_scale_flow.py:96
↓ 1 callersFunctionlearning_rate_schedule
(config,global_placeholders)
src/models/shared/learning_rate_schedule.py:3
↓ 1 callersFunctionpolyak
(params, beta)
src/tf_components/optimizer/openai_adamax.py:7
↓ 1 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/imagenet_32_haar/Validation_data.py:56
↓ 1 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/ffhq_1024_haar/Validation_data.py:56
↓ 1 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/celeba_1024_haar/Validation_data.py:56
↓ 1 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/lsun_bedroom_64_haar/Validation_data.py:59
↓ 1 callersMethodpost_process
(self, rgb, add_dequantization_noise=True)
src/models/imagenet_64_haar/Validation_data.py:56
↓ 1 callersMethodsample_latent
(self, shape, temperature=1.0, truncate=False)
src/models/shared/Normalizing_flow.py:75
↓ 1 callersMethodsample_latent_mcmc
(self, conditioning=None,n_batch=1,temperature=1.0, step_size=0.01,adaptation_steps=1,warmup_steps=3)
src/models/shared/Single_scale_flow.py:89
↓ 1 callersFunctionsolver
(params, loss, config, global_placeholders)
src/models/shared/solver.py:5
Method__call__
(self,x,init=False)
src/tf_components/openai_conv.py:29
Method__call__
(self,x)
src/tf_components/openai_zero_conv.py:27
Method__call__
implement simple way to run layers
src/tf_components/Layer.py:22
Method__call__
(self,x,init=False)
src/tf_components/bijector/Coupling.py:17
Method__call__
default to call forward
src/tf_components/bijector/Bijector.py:8
Method__init__
(self,k_size,n_out,stride,edge_bias,activation='relu', dtype=tf.float32,collection=None,name="openai_conv")
src/tf_components/openai_conv.py:12
Method__init__
(self,k_size,n_out,stride,edge_bias, dtype=tf.float32,collection=None,name="openai_zero_conv")
src/tf_components/openai_zero_conv.py:10
Method__init__
auto scoping for TF 1.0 compatibility collection mechanism for simple layer tracking
src/tf_components/Layer.py:8
Method__init__
(self,collection=None,name='haar_split')
src/tf_components/bijector/Haar_split.py:9
Method__init__
(self,n_steps,width,edge_bias,conditional=False,collection=None,name='mutlistep',**remaining_kwargs)
src/tf_components/bijector/Multi_step.py:8
Method__init__
(self,std_log_scale_factor=3.0, dtype=tf.float32, collection=None, name='Element_shift_scale')
src/tf_components/bijector/Element_shift_scale.py:6
Method__init__
(self, levels, collection=None, name='haar_pyramid')
src/tf_components/bijector/Haar_pyramid.py:11
Method__init__
(self,dtype = tf.float32, collection=None, name='invertible_1x1_matrix')
src/tf_components/bijector/Invertible_1x1.py:100
Method__init__
(self,collection=None,name='haar_squeeze')
src/tf_components/bijector/Haar_squeeze.py:11
Method__init__
(self,collection=None,name='squeeze')
src/tf_components/bijector/Squeeze_2d_half.py:6
Method__init__
(self, width, edge_bias, conditional=False, collection=None,name='step')
src/tf_components/bijector/Step.py:8
Method__init__
(self,k_size,width,edge_bias, dtype=tf.float32,collection=None,name="coupling_res_block")
src/tf_components/bijector/Coupling.py:9
Method__init__
(self, compensate=False, collection=None, name='haar_squeeze_split')
src/tf_components/bijector/Haar_squeeze_split.py:12
Method__init__
(self, dtype=tf.float32, collection=None, name='actNorm')
src/tf_components/bijector/Act_norm.py:6
Method__init__
(self,n_steps,width,split,conditional=False,collection=None,name='mutlistep',**remaining_kwargs)
src/tf_components/bijector/Squeeze_step.py:7
Method__init__
(self,collection=None,name="Layer")
src/tf_components/bijector/Bijector.py:5
Method__init__
(self,batch_override=None,shuffle_repeat=True,partial_level=0)
src/models/imagenet_32_haar/Validation_data.py:6
Method__init__
(self)
src/models/imagenet_32_haar/Conditioning_network.py:5
Method__init__
(self,partial_level=-1)
src/models/imagenet_32_haar/Network_body.py:9
Method__init__
(self,partial_level=0)
src/models/imagenet_32_haar/Training_data.py:6
Method__init__
(self,shape,n_steps,coupling_width,spatial_bias,conditional=False,name='Norm_flow')
src/models/shared/Single_scale_flow.py:15
Method__init__
(self, data_shape, n_squeezes)
src/models/shared/Normalizing_flow.py:5
Method__init__
(self,conditioning_network,partial_level=-1)
src/models/shared/Multi_scale_flow.py:7
Method__init__
(self,batch_override=None,shuffle_repeat=True,partial_level=0)
src/models/ffhq_1024_haar/Validation_data.py:6
Method__init__
(self)
src/models/ffhq_1024_haar/Conditioning_network.py:5
Method__init__
(self,partial_level=-1)
src/models/ffhq_1024_haar/Network_body.py:9
next →1–100 of 227, ranked by callers