Training jammed on epoch 1

JeffWang987 / MVSTER

[ECCV 2022] MVSTER: Epipolar Transformer for Efficient Multi-View Stereo

MIT License

183 stars 13 forks source link

Closed AsDeadAsADodo closed 9 months ago

AsDeadAsADodo commented 9 months ago

AsDeadAsADodo commented 9 months ago

Fixed

add torch.cuda.empty_cache()

on line 128 in train_mvs4.py