00762f3e86
Functional models are way easier to work with, and I don't need any advanced features that would require model subclassing
20 lines
540 B
Python
20 lines
540 B
Python
import tensorflow_datasets as tfds
|
|
|
|
|
|
def load_nyu():
|
|
builder = tfds.builder('nyu_depth_v2')
|
|
builder.download_and_prepare(download_dir='../nyu')
|
|
return builder.as_dataset(split='train', shuffle_files=True)
|
|
|
|
|
|
def print_hi(name):
|
|
# Use a breakpoint in the code line below to debug your script.
|
|
print(f'Hi, {name}') # Press Ctrl+F8 to toggle the breakpoint.
|
|
|
|
|
|
# Press the green button in the gutter to run the script.
|
|
if __name__ == '__main__':
|
|
load_nyu()
|
|
|
|
# See PyCharm help at https://www.jetbrains.com/help/pycharm/
|