can you share your code requirment?

zjjMaiMai / TinyHITNet

HITNet: Hierarchical Iterative Tile Refinement Network for Real-time Stereo Matching

152 stars 21 forks source link

can you share your code requirment? #4

Closed williamhyin closed 2 years ago

williamhyin commented 2 years ago

Hi ,

can you share your code dependency requirement, which could help us alot. Because i have faced a problem:

File "train.py", line 9, in from metrics.epe import EPEMetric File "/workspace/TinyHITNet/metrics/epe.py", line 2, in from torchmetrics import AverageMeter ImportError: cannot import name 'AverageMeter'

even though i have installed pytorch-lightning.

Thanks

williamhyin commented 2 years ago

Hi I found the issue:

https://github.com/PyTorchLightning/metrics/releases

Renamed AverageMeter to MeanMetric (#506)

zjjMaiMai commented 2 years ago

pip install torchmetrics