"train_contrastive_feature.py"报错

[candykilla]

python train_contrastive_feature.py -m output/73d7cf30-b

RFN weight: 1.0 [08/05 21:42:02] RFN weight: 1.0 [08/05 21:42:02] RFN weight: 1.0 [08/05 21:42:02] RFN weight: 1.0 [08/05 21:42:02] Smooth K: 16 [08/05 21:42:02] Scale aware dim: -1 [08/05 21:42:02] Traceback (most recent call last): File "train_contrastive_feature.py", line 369, in training(lp.extract(args), op.extract(args), pp.extract(args), args.iteration, args.save_iterations, args.checkpoint_iterations, args.debug_from) File "train_contrastive_feature.py", line 68, in training assert opt.ray_sample_rate > 0 or opt.num_sampled_rays > 0 AssertionError

[Jumpat]

Oh, you should set the num_sampled_rays or the ray_sample_rate, which are set for reducing GPU memory consumption.

python  train_contrastive_feature.py -m output/73d7cf30-b --num_sampled_rays 1000

I think this should work.