alterzero / starnet Goto Github PK

Can i test my vidoe

IndexError in DataLoader worker process 7...

This is the script of training:

CUDA_VISIBLE_DEVICES=0,1  \
python main_refinement_flow.py \
--batch_size 3\
--gpus 2 \
--threads 30

I faced the same issue while trying training on Dataset of length 785 with corresponding Dataloader with batch size of 8.
Making Dataset length divisible by the batch size solved the issue

Updated training script, works better.

CUDA_VISIBLE_DEVICES=0,1,2 \
python main_refinement_flow.py \
--batch_size 3\
--gpus 3 \
--threads 30

when I use the pretrained model download in the eval program, the parameters don't match as follows, could you check it? Thanks a lot:
RuntimeError: Error(s) in loading state_dict for Net:
Missing key(s) in state_dict: "feat0.conv.weight", "feat0.conv.bias", "feat0.act.weight", "feat1.conv.weight", "feat1.conv.bias", "feat1.act.weight", "DBPN.feat1.conv.weight", "DBPN.feat1.conv.bias", "DBPN.feat1.act.weight", "DBPN.up1.up_conv1.deconv.weight", "DBPN.up1.up_conv1.deconv.bias", "DBPN.up1.up_conv1.act.weight", "DBPN.up1.up_conv2.conv.weight", "DBPN.up1.up_conv2.conv.bias", "DBPN.up1.up_conv2.act.weight", "DBPN.up1.up_conv3.deconv.weight", "DBPN.up1.up_conv3.deconv.bias", "DBPN.up1.up_conv3.act.weight", "DBPN.down1.down_conv1.conv.weight", "DBPN.down1.down_conv1.conv.bias", "DBPN.down1.down_conv1.act.weight", "DBPN.down1.down_conv2.deconv.weight", "DBPN.down1.down_conv2.deconv.bias", "DBPN.down1.down_conv2.act.weight", "DBPN.down1.down_conv3.conv.weight", "DBPN.down1.down_conv3.conv.bias", "DBPN.down1.down_conv3.act.weight", "DBPN.up2.up_conv1.deconv.weight", "DBPN.up2.up_conv1.deconv.bias", "DBPN.up2.up_conv1.act.weight", "DBPN.up2.up_conv2.conv.weight", "DBPN.up2.up_conv2.conv.bias", "DBPN.up2.up_conv2.act.weight", "DBPN.up2.up_conv3.deconv.weight", "DBPN.up2.up_conv3.deconv.bias", "DBPN.up2.up_conv3.act.weight", "DBPN.down2.down_conv1.conv.weight", "DBPN.down2.down_conv1.conv.bias", "DBPN.down2.down_conv1.act.weight", "DBPN.down2.down_conv2.deconv.weight", "DBPN.down2.down_conv2.deconv.bias", "DBPN.down2.down_conv2.act.weight", "DBPN.down2.down_conv3.conv.weight", "DBPN.down2.down_conv3.conv.bias", "DBPN.down2.down_conv3.act.weight", "DBPN.up3.up_conv1.deconv.weight", "DBPN.up3.up_conv1.deconv.bias", "DBPN.up3.up_conv1.act.weight", "DBPN.up3.up_conv2.conv.weight", "DBPN.up3.up_conv2.conv.bias", "DBPN.up3.up_conv2.act.weight", "DBPN.up3.up_conv3.deconv.weight", "DBPN.up3.up_conv3.deconv.bias", "DBPN.up3.up_conv3.act.weight", "DBPN.output.conv.weight", "DBPN.output.conv.bias", "res_feat1.0.conv1.weight", "res_feat1.0.conv1.bias", "res_feat1.0.conv2.weight", "res_feat1.0.conv2.bias", "res_feat1.0.act.weight", "res_feat1.1.conv1.weight", "res_feat1.1.conv1.bias", "res_feat1.1.conv2.weight", "res_feat1.1.conv2.bias", "res_feat1.1.act.weight", "res_feat1.2.conv1.weight", "res_feat1.2.conv1.bias", "res_feat1.2.conv2.weight", "res_feat1.2.conv2.bias", "res_feat1.2.act.weight", "res_feat1.3.conv1.weight", "res_feat1.3.conv1.bias", "res_feat1.3.conv2.weight", "res_feat1.3.conv2.bias", "res_feat1.3.act.weight", "res_feat1.4.conv1.weight", "res_feat1.4.conv1.bias", "res_feat1.4.conv2.weight", "res_feat1.4.conv2.bias", "res_feat1.4.act.weight", "res_feat1.5.deconv.weight", "res_feat1.5.deconv.bias", "res_feat1.5.act.weight", "res_feat2.0.conv1.weight", "res_feat2.0.conv1.bias", "res_feat2.0.conv2.weight", "res_feat2.0.conv2.bias", "res_feat2.0.act.weight", "res_feat2.1.conv1.weight", "res_feat2.1.conv1.bias", "res_feat2.1.conv2.weight", "res_feat2.1.conv2.bias", "res_feat2.1.act.weight", "res_feat2.2.conv1.weight", "res_feat2.2.conv1.bias", "res_feat2.2.conv2.weight", "res_feat2.2.conv2.bias", "res_feat2.2.act.weight", "res_feat2.3.conv1.weight", "res_feat2.3.conv1.bias", "res_feat2.3.conv2.weight", "res_feat2.3.conv2.bias", "res_feat2.3.act.weight", "res_feat2.4.conv1.weight", "res_feat2.4.conv1.bias", "res_feat2.4.conv2.weight", "res_feat2.4.conv2.bias", "res_feat2.4.act.weight", "res_feat2.5.conv.weight", "res_feat2.5.conv.bias", "res_feat2.5.act.weight", "res_feat3.0.conv1.weight", "res_feat3.0.conv1.bias", "res_feat3.0.conv2.weight", "res_feat3.0.conv2.bias", "res_feat3.0.act.weight", "res_feat3.1.conv1.weight", "res_feat3.1.conv1.bias", "res_feat3.1.conv2.weight", "res_feat3.1.conv2.bias", "res_feat3.1.act.weight", "res_feat3.2.conv1.weight", "res_feat3.2.conv1.bias", "res_feat3.2.conv2.weight", "res_feat3.2.conv2.bias", "res_feat3.2.act.weight", "res_feat3.3.conv1.weight", "res_feat3.3.conv1.bias", "res_feat3.3.conv2.weight", "res_feat3.3.conv2.bias", "res_feat3.3.act.weight", "res_feat3.4.conv1.weight", "res_feat3.4.conv1.bias", "res_feat3.4.conv2.weight", "res_feat3.4.conv2.bias", "res_feat3.4.act.weight", "res_feat3.5.conv.weight", "res_feat3.5.conv.bias", "res_feat3.5.act.weight", "output.conv.weight", "output.conv.bias".
Unexpected key(s) in state_dict: "module.model.motion_feat.conv.weight", "module.model.motion_feat.conv.bias", "module.model.motion.0.conv1.weight", "module.model.motion.0.conv1.bias", "module.model.motion.0.conv2.weight", "module.model.motion.0.conv2.bias", "module.model.motion.1.conv1.weight", "module.model.motion.1.conv1.bias", "module.model.motion.1.conv2.weight", "module.model.motion.1.conv2.bias", "module.model.motion.2.conv.weight", "module.model.motion.2.conv.bias", "module.model.t_net_lr.0.conv1.weight", "module.model.t_net_lr.0.conv1.bias", "module.model.t_net_lr.0.conv2.weight", "module.model.t_net_lr.0.conv2.bias", "module.model.t_net_lr.1.conv1.weight", "module.model.t_net_lr.1.conv1.bias", "module.model.t_net_lr.1.conv2.weight", "module.model.t_net_lr.1.conv2.bias", "module.model.t_net_lr.2.conv.weight", "module.model.t_net_lr.2.conv.bias", "module.model.t_net_hr.0.conv.weight", "module.model.t_net_hr.0.conv.bias", "module.model.t_net_hr.1.l1_1.conv1.weight", "module.model.t_net_hr.1.l1_1.conv1.bias", "module.model.t_net_hr.1.l1_1.conv2.weight", "module.model.t_net_hr.1.l1_1.conv2.bias", "module.model.t_net_hr.1.l1_2.conv1.weight", "module.model.t_net_hr.1.l1_2.conv1.bias", "module.model.t_net_hr.1.l1_2.conv2.weight", "module.model.t_net_hr.1.l1_2.conv2.bias", "module.model.t_net_hr.1.l1_3.conv1.weight", "module.model.t_net_hr.1.l1_3.conv1.bias", "module.model.t_net_hr.1.l1_3.conv2.weight", "module.model.t_net_hr.1.l1_3.conv2.bias", "module.model.t_net_hr.1.l1_4.conv1.weight", "module.model.t_net_hr.1.l1_4.conv1.bias", "module.model.t_net_hr.1.l1_4.conv2.weight", "module.model.t_net_hr.1.l1_4.conv2.bias", "module.model.t_net_hr.1.l1_5.conv1.weight", "module.model.t_net_hr.1.l1_5.conv1.bias", "module.model.t_net_hr.1.l1_5.conv2.weight", "module.model.t_net_hr.1.l1_5.conv2.bias", "module.model.t_net_hr.1.l2_1.conv1.weight", "module.model.t_net_hr.1.l2_1.conv1.bias", "module.model.t_net_hr.1.l2_1.conv2.weight", "module.model.t_net_hr.1.l2_1.conv2.bias", "module.model.t_net_hr.1.l2_2.conv1.weight", "module.model.t_net_hr.1.l2_2.conv1.bias", "module.model.t_net_hr.1.l2_2.conv2.weight", "module.model.t_net_hr.1.l2_2.conv2.bias", "module.model.t_net_hr.1.l2_3.conv1.weight", "module.model.t_net_hr.1.l2_3.conv1.bias", "module.model.t_net_hr.1.l2_3.conv2.weight", "module.model.t_net_hr.1.l2_3.conv2.bias", "module.model.t_net_hr.1.l2_4.conv1.weight", "module.model.t_net_hr.1.l2_4.conv1.bias", "module.model.t_net_hr.1.l2_4.conv2.weight", "module.model.t_net_hr.1.l2_4.conv2.bias", "module.model.t_net_hr.1.l3_1.conv1.weight", "module.model.t_net_hr.1.l3_1.conv1.bias", "module.model.t_net_hr.1.l3_1.conv2.weight", "module.model.t_net_hr.1.l3_1.conv2.bias", "module.model.t_net_hr.1.l3_2.conv1.weight", "module.model.t_net_hr.1.l3_2.conv1.bias", "module.model.t_net_hr.1.l3_2.conv2.weight", "module.model.t_net_hr.1.l3_2.conv2.bias", "module.model.t_net_hr.1.l3_3.conv1.weight", "module.model.t_net_hr.1.l3_3.conv1.bias", "module.model.t_net_hr.1.l3_3.conv2.weight", "module.model.t_net_hr.1.l3_3.conv2.bias", "module.model.t_net_hr.1.down1.conv.weight", "module.model.t_net_hr.1.down1.conv.bias", "module.model.t_net_hr.1.down2.conv.weight", "module.model.t_net_hr.1.down2.conv.bias", "module.model.t_net_hr.1.up1.deconv.weight", "module.model.t_net_hr.1.up1.deconv.bias", "module.model.t_net_hr.1.up2.deconv.weight", "module.model.t_net_hr.1.up2.deconv.bias", "module.model.t_net_hr.1.final.conv.weight", "module.model.t_net_hr.1.final.conv.bias", "module.model.t_net_hr.2.conv.weight", "module.model.t_net_hr.2.conv.bias", "module.model.interp_block.0.conv1.weight", "module.model.interp_block.0.conv1.bias", "module.model.interp_block.0.conv2.weight", "module.model.interp_block.0.conv2.bias", "module.model.interp_block.1.conv1.weight", "module.model.interp_block.1.conv1.bias", "module.model.interp_block.1.conv2.weight", "module.model.interp_block.1.conv2.bias", "module.model.interp_block.2.conv1.weight", "module.model.interp_block.2.conv1.bias", "module.model.interp_block.2.conv2.weight", "module.model.interp_block.2.conv2.bias", "module.model.interp_block.3.conv1.weight", "module.model.interp_block.3.conv1.bias", "module.model.interp_block.3.conv2.weight", "module.model.interp_block.3.conv2.bias", "module.model.interp_block.4.conv1.weight", "module.model.interp_block.4.conv1.bias", "module.model.interp_block.4.conv2.weight", "module.model.interp_block.4.conv2.bias", "module.model.interp_block.5.deconv.weight", "module.model.interp_block.5.deconv.bias", "module.model.motion_up_f.0.conv1.weight", "module.model.motion_up_f.0.conv1.bias", "module.model.motion_up_f.0.conv2.weight", "module.model.motion_up_f.0.conv2.bias", "module.model.motion_up_f.1.conv1.weight", "module.model.motion_up_f.1.conv1.bias", "module.model.motion_up_f.1.conv2.weight", "module.model.motion_up_f.1.conv2.bias", "module.model.motion_up_f.2.conv1.weight", "module.model.motion_up_f.2.conv1.bias", "module.model.motion_up_f.2.conv2.weight", "module.model.motion_up_f.2.conv2.bias", "module.model.motion_up_f.3.conv1.weight", "module.model.motion_up_f.3.conv1.bias", "module.model.motion_up_f.3.conv2.weight", "module.model.motion_up_f.3.conv2.bias", "module.model.motion_up_f.4.conv1.weight", "module.model.motion_up_f.4.conv1.bias", "module.model.motion_up_f.4.conv2.weight", "module.model.motion_up_f.4.conv2.bias", "module.model.motion_up_f.5.deconv.weight", "module.model.motion_up_f.5.deconv.bias", "module.model.motion_up_b.0.conv1.weight", "module.model.motion_up_b.0.conv1.bias", "module.model.motion_up_b.0.conv2.weight", "module.model.motion_up_b.0.conv2.bias", "module.model.motion_up_b.1.conv1.weight", "module.model.motion_up_b.1.conv1.bias", "module.model.motion_up_b.1.conv2.weight", "module.model.motion_up_b.1.conv2.bias", "module.model.motion_up_b.2.conv1.weight", "module.model.motion_up_b.2.conv1.bias", "module.model.motion_up_b.2.conv2.weight", "module.model.motion_up_b.2.conv2.bias", "module.model.motion_up_b.3.conv1.weight", "module.model.motion_up_b.3.conv1.bias", "module.model.motion_up_b.3.conv2.weight", "module.model.motion_up_b.3.conv2.bias", "module.model.motion_up_b.4.conv1.weight", "module.model.motion_up_b.4.conv1.bias", "module.model.motion_up_b.4.conv2.weight", "module.model.motion_up_b.4.conv2.bias", "module.model.motion_up_b.5.deconv.weight", "module.model.motion_up_b.5.deconv.bias", "module.model.motion_down.0.conv1.weight", "module.model.motion_down.0.conv1.bias", "module.model.motion_down.0.conv2.weight", "module.model.motion_down.0.conv2.bias", "module.model.motion_down.1.conv1.weight", "module.model.motion_down.1.conv1.bias", "module.model.motion_down.1.conv2.weight", "module.model.motion_down.1.conv2.bias", "module.model.motion_down.2.conv.weight", "module.model.motion_down.2.conv.bias", "module.model.reconstruction_l.conv.weight", "module.model.reconstruction_l.conv.bias", "module.model.reconstruction_h.conv.weight", "module.model.reconstruction_h.conv.bias", "module.model.RBPN.feat0.conv.weight", "module.model.RBPN.feat0.conv.bias", "module.model.RBPN.feat0.act.weight", "module.model.RBPN.feat1.conv.weight", "module.model.RBPN.feat1.conv.bias", "module.model.RBPN.feat1.act.weight", "module.model.RBPN.DBPN.feat1.conv.weight", "module.model.RBPN.DBPN.feat1.conv.bias", "module.model.RBPN.DBPN.feat1.act.weight", "module.model.RBPN.DBPN.up1.up_conv1.deconv.weight", "module.model.RBPN.DBPN.up1.up_conv1.deconv.bias", "module.model.RBPN.DBPN.up1.up_conv1.act.weight", "module.model.RBPN.DBPN.up1.up_conv2.conv.weight", "module.model.RBPN.DBPN.up1.up_conv2.conv.bias", "module.model.RBPN.DBPN.up1.up_conv2.act.weight", "module.model.RBPN.DBPN.up1.up_conv3.deconv.weight", "module.model.RBPN.DBPN.up1.up_conv3.deconv.bias", "module.model.RBPN.DBPN.up1.up_conv3.act.weight", "module.model.RBPN.DBPN.down1.down_conv1.conv.weight", "module.model.RBPN.DBPN.down1.down_conv1.conv.bias", "module.model.RBPN.DBPN.down1.down_conv1.act.weight", "module.model.RBPN.DBPN.down1.down_conv2.deconv.weight", "module.model.RBPN.DBPN.down1.down_conv2.deconv.bias", "module.model.RBPN.DBPN.down1.down_conv2.act.weight", "module.model.RBPN.DBPN.down1.down_conv3.conv.weight", "module.model.RBPN.DBPN.down1.down_conv3.conv.bias", "module.model.RBPN.DBPN.down1.down_conv3.act.weight", "module.model.RBPN.DBPN.up2.up_conv1.deconv.weight", "module.model.RBPN.DBPN.up2.up_conv1.deconv.bias", "module.model.RBPN.DBPN.up2.up_conv1.act.weight", "module.model.RBPN.DBPN.up2.up_conv2.conv.weight", "module.model.RBPN.DBPN.up2.up_conv2.conv.bias", "module.model.RBPN.DBPN.up2.up_conv2.act.weight", "module.model.RBPN.DBPN.up2.up_conv3.deconv.weight", "module.model.RBPN.DBPN.up2.up_conv3.deconv.bias", "module.model.RBPN.DBPN.up2.up_conv3.act.weight", "module.model.RBPN.DBPN.down2.down_conv1.conv.weight", "module.model.RBPN.DBPN.down2.down_conv1.conv.bias", "module.model.RBPN.DBPN.down2.down_conv1.act.weight", "module.model.RBPN.DBPN.down2.down_conv2.deconv.weight", "module.model.RBPN.DBPN.down2.down_conv2.deconv.bias", "module.model.RBPN.DBPN.down2.down_conv2.act.weight", "module.model.RBPN.DBPN.down2.down_conv3.conv.weight", "module.model.RBPN.DBPN.down2.down_conv3.conv.bias", "module.model.RBPN.DBPN.down2.down_conv3.act.weight", "module.model.RBPN.DBPN.up3.up_conv1.deconv.weight", "module.model.RBPN.DBPN.up3.up_conv1.deconv.bias", "module.model.RBPN.DBPN.up3.up_conv1.act.weight", "module.model.RBPN.DBPN.up3.up_conv2.conv.weight", "module.model.RBPN.DBPN.up3.up_conv2.conv.bias", "module.model.RBPN.DBPN.up3.up_conv2.act.weight", "module.model.RBPN.DBPN.up3.up_conv3.deconv.weight", "module.model.RBPN.DBPN.up3.up_conv3.deconv.bias", "module.model.RBPN.DBPN.up3.up_conv3.act.weight", "module.model.RBPN.DBPN.output.conv.weight", "module.model.RBPN.DBPN.output.conv.bias", "module.model.RBPN.res_feat1.0.conv1.weight", "module.model.RBPN.res_feat1.0.conv1.bias", "module.model.RBPN.res_feat1.0.conv2.weight", "module.model.RBPN.res_feat1.0.conv2.bias", "module.model.RBPN.res_feat1.0.act.weight", "module.model.RBPN.res_feat1.1.conv1.weight", "module.model.RBPN.res_feat1.1.conv1.bias", "module.model.RBPN.res_feat1.1.conv2.weight", "module.model.RBPN.res_feat1.1.conv2.bias", "module.model.RBPN.res_feat1.1.act.weight", "module.model.RBPN.res_feat1.2.conv1.weight", "module.model.RBPN.res_feat1.2.conv1.bias", "module.model.RBPN.res_feat1.2.conv2.weight", "module.model.RBPN.res_feat1.2.conv2.bias", "module.model.RBPN.res_feat1.2.act.weight", "module.model.RBPN.res_feat1.3.conv1.weight", "module.model.RBPN.res_feat1.3.conv1.bias", "module.model.RBPN.res_feat1.3.conv2.weight", "module.model.RBPN.res_feat1.3.conv2.bias", "module.model.RBPN.res_feat1.3.act.weight", "module.model.RBPN.res_feat1.4.conv1.weight", "module.model.RBPN.res_feat1.4.conv1.bias", "module.model.RBPN.res_feat1.4.conv2.weight", "module.model.RBPN.res_feat1.4.conv2.bias", "module.model.RBPN.res_feat1.4.act.weight", "module.model.RBPN.res_feat1.5.deconv.weight", "module.model.RBPN.res_feat1.5.deconv.bias", "module.model.RBPN.res_feat1.5.act.weight", "module.model.RBPN.res_feat2.0.conv1.weight", "module.model.RBPN.res_feat2.0.conv1.bias", "module.model.RBPN.res_feat2.0.conv2.weight", "module.model.RBPN.res_feat2.0.conv2.bias", "module.model.RBPN.res_feat2.0.act.weight", "module.model.RBPN.res_feat2.1.conv1.weight", "module.model.RBPN.res_feat2.1.conv1.bias", "module.model.RBPN.res_feat2.1.conv2.weight", "module.model.RBPN.res_feat2.1.conv2.bias", "module.model.RBPN.res_feat2.1.act.weight", "module.model.RBPN.res_feat2.2.conv1.weight", "module.model.RBPN.res_feat2.2.conv1.bias", "module.model.RBPN.res_feat2.2.conv2.weight", "module.model.RBPN.res_feat2.2.conv2.bias", "module.model.RBPN.res_feat2.2.act.weight", "module.model.RBPN.res_feat2.3.conv1.weight", "module.model.RBPN.res_feat2.3.conv1.bias", "module.model.RBPN.res_feat2.3.conv2.weight", "module.model.RBPN.res_feat2.3.conv2.bias", "module.model.RBPN.res_feat2.3.act.weight", "module.model.RBPN.res_feat2.4.conv1.weight", "module.model.RBPN.res_feat2.4.conv1.bias", "module.model.RBPN.res_feat2.4.conv2.weight", "module.model.RBPN.res_feat2.4.conv2.bias", "module.model.RBPN.res_feat2.4.act.weight", "module.model.RBPN.res_feat2.5.conv.weight", "module.model.RBPN.res_feat2.5.conv.bias", "module.model.RBPN.res_feat2.5.act.weight", "module.model.RBPN.res_feat3.0.conv1.weight", "module.model.RBPN.res_feat3.0.conv1.bias", "module.model.RBPN.res_feat3.0.conv2.weight", "module.model.RBPN.res_feat3.0.conv2.bias", "module.model.RBPN.res_feat3.0.act.weight", "module.model.RBPN.res_feat3.1.conv1.weight", "module.model.RBPN.res_feat3.1.conv1.bias", "module.model.RBPN.res_feat3.1.conv2.weight", "module.model.RBPN.res_feat3.1.conv2.bias", "module.model.RBPN.res_feat3.1.act.weight", "module.model.RBPN.res_feat3.2.conv1.weight", "module.model.RBPN.res_feat3.2.conv1.bias", "module.model.RBPN.res_feat3.2.conv2.weight", "module.model.RBPN.res_feat3.2.conv2.bias", "module.model.RBPN.res_feat3.2.act.weight", "module.model.RBPN.res_feat3.3.conv1.weight", "module.model.RBPN.res_feat3.3.conv1.bias", "module.model.RBPN.res_feat3.3.conv2.weight", "module.model.RBPN.res_feat3.3.conv2.bias", "module.model.RBPN.res_feat3.3.act.weight", "module.model.RBPN.res_feat3.4.conv1.weight", "module.model.RBPN.res_feat3.4.conv1.bias", "module.model.RBPN.res_feat3.4.conv2.weight", "module.model.RBPN.res_feat3.4.conv2.bias", "module.model.RBPN.res_feat3.4.act.weight", "module.model.RBPN.res_feat3.5.conv.weight", "module.model.RBPN.res_feat3.5.conv.bias", "module.model.RBPN.res_feat3.5.act.weight", "module.model.RBPN.output.conv.weight", "module.model.RBPN.output.conv.bias", "module.flow_net.inc.conv.conv.0.weight", "module.flow_net.inc.conv.conv.0.bias", "module.flow_net.inc.conv.conv.1.weight", "module.flow_net.inc.conv.conv.1.bias", "module.flow_net.inc.conv.conv.1.running_mean", "module.flow_net.inc.conv.conv.1.running_var", "module.flow_net.inc.conv.conv.1.num_batches_tracked", "module.flow_net.inc.conv.conv.3.weight", "module.flow_net.inc.conv.conv.3.bias", "module.flow_net.inc.conv.conv.4.weight", "module.flow_net.inc.conv.conv.4.bias", "module.flow_net.inc.conv.conv.4.running_mean", "module.flow_net.inc.conv.conv.4.running_var", "module.flow_net.inc.conv.conv.4.num_batches_tracked", "module.flow_net.down1.mpconv.1.conv.0.weight", "module.flow_net.down1.mpconv.1.conv.0.bias", "module.flow_net.down1.mpconv.1.conv.1.weight", "module.flow_net.down1.mpconv.1.conv.1.bias", "module.flow_net.down1.mpconv.1.conv.1.running_mean", "module.flow_net.down1.mpconv.1.conv.1.running_var", "module.flow_net.down1.mpconv.1.conv.1.num_batches_tracked", "module.flow_net.down1.mpconv.1.conv.3.weight", "module.flow_net.down1.mpconv.1.conv.3.bias", "module.flow_net.down1.mpconv.1.conv.4.weight", "module.flow_net.down1.mpconv.1.conv.4.bias", "module.flow_net.down1.mpconv.1.conv.4.running_mean", "module.flow_net.down1.mpconv.1.conv.4.running_var", "module.flow_net.down1.mpconv.1.conv.4.num_batches_tracked", "module.flow_net.down2.mpconv.1.conv.0.weight", "module.flow_net.down2.mpconv.1.conv.0.bias", "module.flow_net.down2.mpconv.1.conv.1.weight", "module.flow_net.down2.mpconv.1.conv.1.bias", "module.flow_net.down2.mpconv.1.conv.1.running_mean", "module.flow_net.down2.mpconv.1.conv.1.running_var", "module.flow_net.down2.mpconv.1.conv.1.num_batches_tracked", "module.flow_net.down2.mpconv.1.conv.3.weight", "module.flow_net.down2.mpconv.1.conv.3.bias", "module.flow_net.down2.mpconv.1.conv.4.weight", "module.flow_net.down2.mpconv.1.conv.4.bias", "module.flow_net.down2.mpconv.1.conv.4.running_mean", "module.flow_net.down2.mpconv.1.conv.4.running_var", "module.flow_net.down2.mpconv.1.conv.4.num_batches_tracked", "module.flow_net.down3.mpconv.1.conv.0.weight", "module.flow_net.down3.mpconv.1.conv.0.bias", "module.flow_net.down3.mpconv.1.conv.1.weight", "module.flow_net.down3.mpconv.1.conv.1.bias", "module.flow_net.down3.mpconv.1.conv.1.running_mean", "module.flow_net.down3.mpconv.1.conv.1.running_var", "module.flow_net.down3.mpconv.1.conv.1.num_batches_tracked", "module.flow_net.down3.mpconv.1.conv.3.weight", "module.flow_net.down3.mpconv.1.conv.3.bias", "module.flow_net.down3.mpconv.1.conv.4.weight", "module.flow_net.down3.mpconv.1.conv.4.bias", "module.flow_net.down3.mpconv.1.conv.4.running_mean", "module.flow_net.down3.mpconv.1.conv.4.running_var", "module.flow_net.down3.mpconv.1.conv.4.num_batches_tracked", "module.flow_net.down4.mpconv.1.conv.0.weight", "module.flow_net.down4.mpconv.1.conv.0.bias", "module.flow_net.down4.mpconv.1.conv.1.weight", "module.flow_net.down4.mpconv.1.conv.1.bias", "module.flow_net.down4.mpconv.1.conv.1.running_mean", "module.flow_net.down4.mpconv.1.conv.1.running_var", "module.flow_net.down4.mpconv.1.conv.1.num_batches_tracked", "module.flow_net.down4.mpconv.1.conv.3.weight", "module.flow_net.down4.mpconv.1.conv.3.bias", "module.flow_net.down4.mpconv.1.conv.4.weight", "module.flow_net.down4.mpconv.1.conv.4.bias", "module.flow_net.down4.mpconv.1.conv.4.running_mean", "module.flow_net.down4.mpconv.1.conv.4.running_var", "module.flow_net.down4.mpconv.1.conv.4.num_batches_tracked", "module.flow_net.down5.mpconv.1.conv.0.weight", "module.flow_net.down5.mpconv.1.conv.0.bias", "module.flow_net.down5.mpconv.1.conv.1.weight", "module.flow_net.down5.mpconv.1.conv.1.bias", "module.flow_net.down5.mpconv.1.conv.1.running_mean", "module.flow_net.down5.mpconv.1.conv.1.running_var", "module.flow_net.down5.mpconv.1.conv.1.num_batches_tracked", "module.flow_net.down5.mpconv.1.conv.3.weight", "module.flow_net.down5.mpconv.1.conv.3.bias", "module.flow_net.down5.mpconv.1.conv.4.weight", "module.flow_net.down5.mpconv.1.conv.4.bias", "module.flow_net.down5.mpconv.1.conv.4.running_mean", "module.flow_net.down5.mpconv.1.conv.4.running_var", "module.flow_net.down5.mpconv.1.conv.4.num_batches_tracked", "module.flow_net.up1.conv.conv.0.weight", "module.flow_net.up1.conv.conv.0.bias", "module.flow_net.up1.conv.conv.1.weight", "module.flow_net.up1.conv.conv.1.bias", "module.flow_net.up1.conv.conv.1.running_mean", "module.flow_net.up1.conv.conv.1.running_var", "module.flow_net.up1.conv.conv.1.num_batches_tracked", "module.flow_net.up1.conv.conv.3.weight", "module.flow_net.up1.conv.conv.3.bias", "module.flow_net.up1.conv.conv.4.weight", "module.flow_net.up1.conv.conv.4.bias", "module.flow_net.up1.conv.conv.4.running_mean", "module.flow_net.up1.conv.conv.4.running_var", "module.flow_net.up1.conv.conv.4.num_batches_tracked", "module.flow_net.up2.conv.conv.0.weight", "module.flow_net.up2.conv.conv.0.bias", "module.flow_net.up2.conv.conv.1.weight", "module.flow_net.up2.conv.conv.1.bias", "module.flow_net.up2.conv.conv.1.running_mean", "module.flow_net.up2.conv.conv.1.running_var", "module.flow_net.up2.conv.conv.1.num_batches_tracked", "module.flow_net.up2.conv.conv.3.weight", "module.flow_net.up2.conv.conv.3.bias", "module.flow_net.up2.conv.conv.4.weight", "module.flow_net.up2.conv.conv.4.bias", "module.flow_net.up2.conv.conv.4.running_mean", "module.flow_net.up2.conv.conv.4.running_var", "module.flow_net.up2.conv.conv.4.num_batches_tracked", "module.flow_net.up3.conv.conv.0.weight", "module.flow_net.up3.conv.conv.0.bias", "module.flow_net.up3.conv.conv.1.weight", "module.flow_net.up3.conv.conv.1.bias", "module.flow_net.up3.conv.conv.1.running_mean", "module.flow_net.up3.conv.conv.1.running_var", "module.flow_net.up3.conv.conv.1.num_batches_tracked", "module.flow_net.up3.conv.conv.3.weight", "module.flow_net.up3.conv.conv.3.bias", "module.flow_net.up3.conv.conv.4.weight", "module.flow_net.up3.conv.conv.4.bias", "module.flow_net.up3.conv.conv.4.running_mean", "module.flow_net.up3.conv.conv.4.running_var", "module.flow_net.up3.conv.conv.4.num_batches_tracked", "module.flow_net.up4.conv.conv.0.weight", "module.flow_net.up4.conv.conv.0.bias", "module.flow_net.up4.conv.conv.1.weight", "module.flow_net.up4.conv.conv.1.bias", "module.flow_net.up4.conv.conv.1.running_mean", "module.flow_net.up4.conv.conv.1.running_var", "module.flow_net.up4.conv.conv.1.num_batches_tracked", "module.flow_net.up4.conv.conv.3.weight", "module.flow_net.up4.conv.conv.3.bias", "module.flow_net.up4.conv.conv.4.weight", "module.flow_net.up4.conv.conv.4.bias", "module.flow_net.up4.conv.conv.4.running_mean", "module.flow_net.up4.conv.conv.4.running_var", "module.flow_net.up4.conv.conv.4.num_batches_tracked", "module.flow_net.up5.conv.conv.0.weight", "module.flow_net.up5.conv.conv.0.bias", "module.flow_net.up5.conv.conv.1.weight", "module.flow_net.up5.conv.conv.1.bias", "module.flow_net.up5.conv.conv.1.running_mean", "module.flow_net.up5.conv.conv.1.running_var", "module.flow_net.up5.conv.conv.1.num_batches_tracked", "module.flow_net.up5.conv.conv.3.weight", "module.flow_net.up5.conv.conv.3.bias", "module.flow_net.up5.conv.conv.4.weight", "module.flow_net.up5.conv.conv.4.bias", "module.flow_net.up5.conv.conv.4.running_mean", "module.flow_net.up5.conv.conv.4.running_var", "module.flow_net.up5.conv.conv.4.num_batches_tracked", "module.flow_net.outc.conv.weight", "module.flow_net.outc.conv.bias".

Process finished with exit code 1

where is rbpn_pretrained_F2_2x.pth?

Hi, your work seems interesting，when will you release your code and paper?

Thank you for your outstanding work,
I want to know max_flow=150, the function of this parameter and how to determine it, thank you very much！

I have install pyflow successfully
But still get this error
"AttributeError: module 'pyflow' has no attribute 'coarse2fine_flow'"

As your instruction
After I run this command "python setup.py build_ext -i" success
I can't run "cp pyflow*.so .."
I got this error
"cp: cannot stat 'pyflow*.so': No such file or directory"

thank you so much!

Hello！Will Adamax be better than Adam?