Function
load_superres_data
(data_dir, batch_size, large_size, small_size, class_cond=False)
Source from the content-addressed store, hash-verified
| 61 | |
| 62 | |
| 63 | def load_superres_data(data_dir, batch_size, large_size, small_size, class_cond=False): |
| 64 | data = load_data( |
| 65 | data_dir=data_dir, |
| 66 | batch_size=batch_size, |
| 67 | image_size=large_size, |
| 68 | class_cond=class_cond, |
| 69 | ) |
| 70 | for large_batch, model_kwargs in data: |
| 71 | model_kwargs["low_res"] = F.interpolate(large_batch, small_size, mode="area") |
| 72 | yield large_batch, model_kwargs |
| 73 | |
| 74 | |
| 75 | def create_argparser(): |
Tested by
no test coverage detected