how to set batch_size during training

yerfor / GeneFacePlusPlus

GeneFace++: Generalized and Stable Real-Time 3D Talking Face Generation; Official Code

MIT License

1.59k stars 229 forks source link

Closed jinqinn closed 8 months ago

jinqinn commented 8 months ago

how to set batch_size during training

pxssw commented 8 months ago

how to set batch_size during training

./tasks/radnerfs/dataset_utils.py", line 440, in collater assert len(samples) == 1 # NeRF only take 1 image for each iteration

jinqinn commented 8 months ago

@pxssw thanks