SiCheng Li

I have experience with model compression and I want to contrbute. I want to know where to get the training dataset and pretrained weights?

**Task** (what are you trying to do/register?) I have many pair of 2D images (each pair only have 2 images: fixed image and moving image) and I want put it...

微信二维码无法添加好友

3

你好，我打算为项目贡献一份力量，但是留下的联系方式无法添加，请查看

Support h5 weight loading in PyTorch training script

1

Adds the ability to load initial weights from a Keras h5 model file when the --load-model argument contains the 'h5' substring. The main changes include: - Check if the `--load-model`...