()
| 22 | return config |
| 23 | |
| 24 | def get_data_config(): |
| 25 | config = ml_collections.ConfigDict() |
| 26 | config.data_rootdir = 'examples' |
| 27 | |
| 28 | config.sample_size = '384x672' |
| 29 | config.dilate_width = 11 |
| 30 | config.max_video_length = 197 |
| 31 | config.fps = 12 |
| 32 | return config |
| 33 | |
| 34 | |
| 35 | def get_video_model_config(): |