import pickle import numpy as np import better_exceptions better_exceptions.hook() import os os.environ['TF_KERAS'] = '1' import os, sys sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')) from ekg.utils.train_utils import allow_gpu_growth allow_gpu_growth() from ekg.utils.train_utils import set_wandb_config # for loging result import wandb from wandb.keras import WandbCallback from tensorflow import keras from tensorflow.keras.optimizers import Adam from keras_radam import RAdam from tensorflow.keras.callbacks import EarlyStopping, ReduceLROnPlateau from ekg.callbacks import LogBest from ekg.utils import data_utils from ekg.utils.data_utils import BaseDataGenerator from ekg.utils.datasets import BigExamLoader, Audicor10sLoader
#!/usr/bin/env python3 import pickle import numpy as np import better_exceptions; better_exceptions.hook() import os os.environ['TF_KERAS'] = '1' import os, sys sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..')) from ekg.utils.train_utils import allow_gpu_growth; allow_gpu_growth() from ekg.utils.train_utils import set_wandb_config # for loging result import wandb from wandb.keras import WandbCallback from tensorflow import keras from tensorflow.keras.optimizers import Adam from keras_radam import RAdam from tensorflow.keras.callbacks import EarlyStopping, ReduceLROnPlateau from ekg.callbacks import LogBest from ekg.utils import data_utils from ekg.utils.data_utils import BaseDataGenerator from ekg.utils.datasets import BigExamLoader, Audicor10sLoader import evaluation from evaluation import print_statistics from ekg.models.backbone import backbone