[Feature] Add a config of TOFlow #811
Conversation
Codecov Report
@@ Coverage Diff @@
## master #811 +/- ##
==========================================
+ Coverage 83.04% 83.08% +0.03%
==========================================
Files 216 219 +3
Lines 12239 12354 +115
Branches 1975 2000 +25
==========================================
+ Hits 10164 10264 +100
- Misses 1767 1774 +7
- Partials 308 316 +8
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
| ] | ||
| metrics[key] = dict( | ||
| PSNR=metrics_data[0], SSIM=metrics_data[1]) | ||
| except ValueError: |
There was a problem hiding this comment.
What will trigger a ValueError?
There was a problem hiding this comment.
Supplementary information for the model, such as Pretrained SPyNet in TOFlow.
| # train | ||
| train=dict( | ||
| type='RepeatDataset', | ||
| times=1000, |
There was a problem hiding this comment.
why repeat 1000 times, instead of let IterBasedRunner iter over it internally?
There was a problem hiding this comment.
This is a safer approach.
There was a problem hiding this comment.
@wangruohui When "reloading" the dataset, it consumes additional time.
| root_dir = 'data/vimeo_triplet' | ||
| data = dict( | ||
| workers_per_gpu=1, | ||
| train_dataloader=dict(samples_per_gpu=1, drop_last=True), |
There was a problem hiding this comment.
How many GPUs by default does this config run on?
There was a problem hiding this comment.
One GPU is enough.
| interval=100, hooks=[ | ||
| dict(type='TextLoggerHook', by_epoch=False), | ||
| ]) | ||
| visual_config = None |
* [Feature] Add config of TOFlow * Update * Update * Update * Update * Update * Update * Update
* [Feature] Add config of TOFlow * Update * Update * Update * Update * Update * Update * Update
No description provided.