Advertisement
Guest User

Untitled

a guest
Nov 6th, 2020
272
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 15.93 KB | None | 0 0
  1. ValueError Traceback (most recent call last)
  2. <ipython-input-17-ab5eae0dd51a> in <module>
  3. 3 ]
  4. 4
  5. ----> 5 run_nuts(
  6. 6 target_log_prob_fn,
  7. 7 inits[:-1]
  8.  
  9. <ipython-input-13-2d7a920574a8> in run_nuts_template(trace_fn, target_log_prob_fn, inits, bijectors_list, num_steps, num_burnin, n_chains)
  10. 45 )
  11. 46
  12. ---> 47 res = tfp.mcmc.sample_chain(
  13. 48 num_results=num_steps,
  14. 49 num_burnin_steps=num_burnin,
  15.  
  16. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/sample.py in sample_chain(num_results, current_state, previous_kernel_results, kernel, num_burnin_steps, num_steps_between_results, trace_fn, return_final_kernel_results, parallel_iterations, seed, name)
  17. 359 return seed, next_state, current_kernel_results
  18. 360
  19. --> 361 (_, _, final_kernel_results), (all_states, trace) = mcmc_util.trace_scan(
  20. 362 loop_fn=_trace_scan_fn,
  21. 363 initial_state=(seed, current_state, previous_kernel_results),
  22.  
  23. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/util.py in trace_scan(loop_fn, initial_state, elems, trace_fn, trace_criterion_fn, static_trace_allocation_size, parallel_iterations, name)
  24. 462 return i + 1, state, num_steps_traced, trace_arrays
  25. 463
  26. --> 464 _, final_state, _, trace_arrays = tf.while_loop(
  27. 465 cond=lambda i, *_: i < length,
  28. 466 body=_body,
  29.  
  30. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py in new_func(*args, **kwargs)
  31. 603 func.__module__, arg_name, arg_value, 'in a future version'
  32. 604 if date is None else ('after %s' % date), instructions)
  33. --> 605 return func(*args, **kwargs)
  34. 606
  35. 607 doc = _add_deprecated_arg_value_notice_to_docstring(
  36.  
  37. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop_v2(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, maximum_iterations, name)
  38. 2487
  39. 2488 """
  40. -> 2489 return while_loop(
  41. 2490 cond=cond,
  42. 2491 body=body,
  43.  
  44. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, name, maximum_iterations, return_same_structure)
  45. 2733 list(loop_vars))
  46. 2734 while cond(*loop_vars):
  47. -> 2735 loop_vars = body(*loop_vars)
  48. 2736 if try_to_pack and not isinstance(loop_vars, (list, _basetuple)):
  49. 2737 packed = True
  50.  
  51. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/util.py in _body(i, state, num_steps_traced, trace_arrays)
  52. 452 def _body(i, state, num_steps_traced, trace_arrays):
  53. 453 elem = elems_array.read(i)
  54. --> 454 state = loop_fn(state, elem)
  55. 455
  56. 456 trace_arrays, num_steps_traced = ps.cond(
  57.  
  58. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/sample.py in _trace_scan_fn(seed_state_and_results, num_steps)
  59. 352
  60. 353 def _trace_scan_fn(seed_state_and_results, num_steps):
  61. --> 354 seed, next_state, current_kernel_results = mcmc_util.smart_for_loop(
  62. 355 loop_num_iter=num_steps,
  63. 356 body_fn=_seeded_one_step,
  64.  
  65. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/util.py in smart_for_loop(loop_num_iter, body_fn, initial_loop_vars, parallel_iterations, unroll_threshold, name)
  66. 346 # where while/LoopCond needs it.
  67. 347 loop_num_iter = tf.cast(loop_num_iter, dtype=tf.int32)
  68. --> 348 return tf.while_loop(
  69. 349 cond=lambda i, *args: i < loop_num_iter,
  70. 350 body=lambda i, *args: [i + 1] + list(body_fn(*args)),
  71.  
  72. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py in new_func(*args, **kwargs)
  73. 603 func.__module__, arg_name, arg_value, 'in a future version'
  74. 604 if date is None else ('after %s' % date), instructions)
  75. --> 605 return func(*args, **kwargs)
  76. 606
  77. 607 doc = _add_deprecated_arg_value_notice_to_docstring(
  78.  
  79. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop_v2(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, maximum_iterations, name)
  80. 2487
  81. 2488 """
  82. -> 2489 return while_loop(
  83. 2490 cond=cond,
  84. 2491 body=body,
  85.  
  86. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, name, maximum_iterations, return_same_structure)
  87. 2733 list(loop_vars))
  88. 2734 while cond(*loop_vars):
  89. -> 2735 loop_vars = body(*loop_vars)
  90. 2736 if try_to_pack and not isinstance(loop_vars, (list, _basetuple)):
  91. 2737 packed = True
  92.  
  93. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/util.py in <lambda>(i, *args)
  94. 348 return tf.while_loop(
  95. 349 cond=lambda i, *args: i < loop_num_iter,
  96. --> 350 body=lambda i, *args: [i + 1] + list(body_fn(*args)),
  97. 351 loop_vars=[np.int32(0)] + initial_loop_vars,
  98. 352 parallel_iterations=parallel_iterations
  99.  
  100. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/sample.py in _seeded_one_step(seed, *state_and_results)
  101. 349 one_step_kwargs = dict(seed=step_seed) if is_seeded else {}
  102. 350 return [passalong_seed] + list(
  103. --> 351 kernel.one_step(*state_and_results, **one_step_kwargs))
  104. 352
  105. 353 def _trace_scan_fn(seed_state_and_results, num_steps):
  106.  
  107. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/dual_averaging_step_size_adaptation.py in one_step(self, current_state, previous_kernel_results, seed)
  108. 454 # Step the inner kernel.
  109. 455 inner_kwargs = {} if seed is None else dict(seed=seed)
  110. --> 456 new_state, new_inner_results = self.inner_kernel.one_step(
  111. 457 current_state, inner_results, **inner_kwargs)
  112. 458
  113.  
  114. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/transformed_kernel.py in one_step(self, current_state, previous_kernel_results, seed)
  115. 397 self.name, 'transformed_kernel', 'one_step')):
  116. 398 inner_kwargs = {} if seed is None else dict(seed=seed)
  117. --> 399 transformed_next_state, kernel_results = self._inner_kernel.one_step(
  118. 400 previous_kernel_results.transformed_state,
  119. 401 previous_kernel_results.inner_results,
  120.  
  121. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/nuts.py in one_step(self, current_state, previous_kernel_results, seed)
  122. 392 )
  123. 393
  124. --> 394 _, _, _, new_step_metastate = tf.while_loop(
  125. 395 cond=lambda iter_, seed, state, metastate: ( # pylint: disable=g-long-lambda
  126. 396 (iter_ < self.max_tree_depth) &
  127.  
  128. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py in new_func(*args, **kwargs)
  129. 603 func.__module__, arg_name, arg_value, 'in a future version'
  130. 604 if date is None else ('after %s' % date), instructions)
  131. --> 605 return func(*args, **kwargs)
  132. 606
  133. 607 doc = _add_deprecated_arg_value_notice_to_docstring(
  134.  
  135. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop_v2(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, maximum_iterations, name)
  136. 2487
  137. 2488 """
  138. -> 2489 return while_loop(
  139. 2490 cond=cond,
  140. 2491 body=body,
  141.  
  142. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, name, maximum_iterations, return_same_structure)
  143. 2733 list(loop_vars))
  144. 2734 while cond(*loop_vars):
  145. -> 2735 loop_vars = body(*loop_vars)
  146. 2736 if try_to_pack and not isinstance(loop_vars, (list, _basetuple)):
  147. 2737 packed = True
  148.  
  149. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/nuts.py in <lambda>(iter_, seed, state, metastate)
  150. 396 (iter_ < self.max_tree_depth) &
  151. 397 tf.reduce_any(metastate.continue_tree)),
  152. --> 398 body=lambda iter_, seed, state, metastate: self._loop_tree_doubling( # pylint: disable=g-long-lambda
  153. 399 previous_kernel_results.step_size,
  154. 400 previous_kernel_results.momentum_state_memory,
  155.  
  156. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/nuts.py in _loop_tree_doubling(self, step_size, momentum_state_memory, current_step_meta_info, iter_, initial_step_state, initial_step_metastate, seed)
  157. 570 momentum_subtree_cumsum,
  158. 571 leapfrogs_taken
  159. --> 572 ] = self._build_sub_tree(
  160. 573 directions_expanded,
  161. 574 integrator,
  162.  
  163. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/nuts.py in _build_sub_tree(self, directions, integrator, current_step_meta_info, nsteps, initial_state, continue_tree, not_divergence, momentum_state_memory, seed, name)
  164. 750 final_not_divergence,
  165. 751 momentum_state_memory,
  166. --> 752 ] = tf.while_loop(
  167. 753 cond=lambda iter_, seed, energy_diff_sum, init_momentum_cumsum, # pylint: disable=g-long-lambda
  168. 754 leapfrogs_taken, state, state_c, continue_tree,
  169.  
  170. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py in new_func(*args, **kwargs)
  171. 603 func.__module__, arg_name, arg_value, 'in a future version'
  172. 604 if date is None else ('after %s' % date), instructions)
  173. --> 605 return func(*args, **kwargs)
  174. 606
  175. 607 doc = _add_deprecated_arg_value_notice_to_docstring(
  176.  
  177. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop_v2(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, maximum_iterations, name)
  178. 2487
  179. 2488 """
  180. -> 2489 return while_loop(
  181. 2490 cond=cond,
  182. 2491 body=body,
  183.  
  184. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, name, maximum_iterations, return_same_structure)
  185. 2733 list(loop_vars))
  186. 2734 while cond(*loop_vars):
  187. -> 2735 loop_vars = body(*loop_vars)
  188. 2736 if try_to_pack and not isinstance(loop_vars, (list, _basetuple)):
  189. 2737 packed = True
  190.  
  191. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/nuts.py in <lambda>(iter_, seed, energy_diff_sum, init_momentum_cumsum, leapfrogs_taken, state, state_c, continue_tree, not_divergence, momentum_state_memory)
  192. 758 leapfrogs_taken, state, state_c, continue_tree,
  193. 759 not_divergence, momentum_state_memory: (
  194. --> 760 self._loop_build_sub_tree(
  195. 761 directions, integrator, current_step_meta_info,
  196. 762 iter_, energy_diff_sum, init_momentum_cumsum,
  197.  
  198. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/nuts.py in _loop_build_sub_tree(self, directions, integrator, current_step_meta_info, iter_, energy_diff_sum_previous, momentum_cumsum_previous, leapfrogs_taken, prev_tree_state, candidate_tree_state, continue_tree_previous, not_divergent_previous, momentum_state_memory, seed)
  199. 811 next_target,
  200. 812 next_target_grad_parts
  201. --> 813 ] = integrator(prev_tree_state.momentum,
  202. 814 prev_tree_state.state,
  203. 815 prev_tree_state.target,
  204.  
  205. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/leapfrog_integrator.py in __call__(self, momentum_parts, state_parts, target, target_grad_parts, kinetic_energy_fn, name)
  206. 295 next_target,
  207. 296 next_target_grad_parts,
  208. --> 297 ] = tf.while_loop(
  209. 298 cond=lambda i, *_: i < self.num_steps,
  210. 299 body=lambda i, *args: [i + 1] + list(_one_step( # pylint: disable=no-value-for-parameter,g-long-lambda
  211.  
  212. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/util/deprecation.py in new_func(*args, **kwargs)
  213. 603 func.__module__, arg_name, arg_value, 'in a future version'
  214. 604 if date is None else ('after %s' % date), instructions)
  215. --> 605 return func(*args, **kwargs)
  216. 606
  217. 607 doc = _add_deprecated_arg_value_notice_to_docstring(
  218.  
  219. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop_v2(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, maximum_iterations, name)
  220. 2487
  221. 2488 """
  222. -> 2489 return while_loop(
  223. 2490 cond=cond,
  224. 2491 body=body,
  225.  
  226. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/ops/control_flow_ops.py in while_loop(cond, body, loop_vars, shape_invariants, parallel_iterations, back_prop, swap_memory, name, maximum_iterations, return_same_structure)
  227. 2733 list(loop_vars))
  228. 2734 while cond(*loop_vars):
  229. -> 2735 loop_vars = body(*loop_vars)
  230. 2736 if try_to_pack and not isinstance(loop_vars, (list, _basetuple)):
  231. 2737 packed = True
  232.  
  233. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/leapfrog_integrator.py in <lambda>(i, *args)
  234. 297 ] = tf.while_loop(
  235. 298 cond=lambda i, *_: i < self.num_steps,
  236. --> 299 body=lambda i, *args: [i + 1] + list(_one_step( # pylint: disable=no-value-for-parameter,g-long-lambda
  237. 300 self.target_fn, self.step_sizes, get_velocity_parts, *args)),
  238. 301 loop_vars=[
  239.  
  240. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/mcmc/internal/leapfrog_integrator.py in _one_step(target_fn, step_sizes, get_velocity_parts, half_next_momentum_parts, state_parts, target, target_grad_parts)
  241. 353 next_target_grad_parts))
  242. 354
  243. --> 355 tensorshape_util.set_shape(next_target, target.shape)
  244. 356 for ng, g in zip(next_target_grad_parts, target_grad_parts):
  245. 357 tensorshape_util.set_shape(ng, g.shape)
  246.  
  247. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow_probability/python/internal/tensorshape_util.py in set_shape(tensor, shape)
  248. 326 """
  249. 327 if hasattr(tensor, 'set_shape'):
  250. --> 328 tensor.set_shape(shape)
  251. 329
  252. 330
  253.  
  254. ~/anaconda3/envs/tensor/lib/python3.8/site-packages/tensorflow/python/framework/ops.py in set_shape(self, shape)
  255. 1213 def set_shape(self, shape):
  256. 1214 if not self.shape.is_compatible_with(shape):
  257. -> 1215 raise ValueError(
  258. 1216 "Tensor's shape %s is not compatible with supplied shape %s" %
  259. 1217 (self.shape, shape))
  260.  
  261. ValueError: Tensor's shape (2, 2) is not compatible with supplied shape (2,)
  262.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement