Add metrics, prepare for training

main.py

@@ -7,13 +7,5 @@ def load_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/