hk.module must be initialized inside an hk.transform

临时解决方法：

  def test_output_dimension(self, final_endpoint, expected_shape):input_shape = (2, 32, 224, 224, 3)def f():data = jnp.zeros(input_shape)net = tsm_resnet.TSMResNetV2()return net(data, final_endpoint=final_endpoint)init_fn, apply_fn = hk.transform(f)out = apply_fn(init_fn(jax.random.PRNGKey(42)), None)self.assertEqual(out.shape, expected_shape)

参考：

https://ask.csdn.net/questions/1221077

python
class Parameter():def __init__(self, init_value: float, name: Text):super().__init__(name="parameter")self._name = nameself._init = hk.initializer.Constant(jnp.log(init_value))def __call__(self):return jnp.exp(hk.get_para

hk.module must be initialized inside an hk.transform相关推荐

hk域名如何注册？为什么要选择.hk域名？
hk域名属于香港的一所互联网公司,该域名不仅不需要收费,而且注册要求不高,因此深受各大用户的认可.那么hk域名如何注册?为什么要选择.hk域名?我们和大家说一下注册的技巧! 一.为什么要选择.hk域名 ...
南方东英恒生科技指数每日杠杆(2x)产品（股票代码： 7226.hk）和南方东英恒生科技指数每日反向(-2x) 产品（股票代码： 7552.hk）将在港交所上市
香港 -- (美国商业资讯) -- 在南方东英恒生科技指数ETF(股票代码:3033.hk)取得巨大成功的基础上,南方东英资产管理有限公司("南方东英")再度推出一对追踪恒生科技指 ...
Center and Scale Prediction: A Box-free Approach for Object Detection
Center and Scale Prediction: A Box-free Approach for Object Detection 中心和尺度预测:一种 box-free 的物体检测方法无锚 ...
DeepMind发布神经网络、强化学习库，网友：推动JAX发展
点击上方"AI遇见机器学习",选择"星标"公众号重磅干货,第一时间送达十三发自凹非寺量子位报道 | 公众号 QbitAI DeepMind今日发布了 ...
dm-haiku 用法
pip install jax jaxlib dm-haiku,2021.02.16只支持linux class MyDropout(hk.Module):def __init__(self, rat ...
激光IMU融合——LIO-Mapping / LIOM / LINS / LIO-SAM算法解析
激光IMU融合--LIO-Mapping / LIOM / LINS / LIO-SAM算法解析激光IMU融合--LIO-Mapping / LIOM / LINS / LIO-SAM算法解析 1. ...
smartforms句柄与以簇的方式存储数据。
说明:Smartforms输出打印,设置句柄,用于传递参数. REPORT ZIEB_WWFLD. *----------------- TABLES ------------- ...
vue-cli2、vue-cli3脚手架详细讲解
转载自 vue-cli2.vue-cli3脚手架详细讲解前言: vue脚手架指的是vue-cli它是vue官方提供的一个快速构建单页面(SPA)环境配置的工具,cli 就是(command-line ...
dagger2 注入_如何使用Dagger 2在您的应用程序中实现依赖注入
dagger2 注入 Kriptofolio应用程序系列-第4部分 (Kriptofolio app series - Part 4) Dependency injection will signif ...

hk.module must be initialized inside an hk.transform

hk.module must be initialized inside an hk.transform相关推荐

最新文章

热门文章