import sys
import logging
from zipfile import ZipFile
import tensorflow as tf
import tensorflow_datasets as tfds
import bdlb
from tqdm import tqdm
import time
from .utils import ExperimentData
from fs.data.fsdata import FSData
from fs.data.utils import load_gdrive_file
from fs.data.augmentation import crop_multiple
ex = Experiment()
ex.capture_out_filter = apply_backspaces_and_linefeeds
ex.observers.append(get_observer())
@ex.command
def saved_model(testing_dataset, model_id, _run, _log, batching=False, validation=False):
data = tfds.load(name='cityscapes', split='validation',
data_dir='/cluster/work/riner/users/blumh/tensorflow_datasets')
if batching:
data = data.batch(1)
data = data.prefetch(500)
ZipFile(load_gdrive_file(model_id, 'zip')).extractall('/tmp/extracted_module')
tf.compat.v1.enable_resource_variables()
net = tf.saved_model.load('/tmp/extracted_module')
