Juartaurus's picture
Upload folder using huggingface_hub
1865436
raw
history blame
726 Bytes
import sys
sys.path.insert(0, 'src/ir')
import dotenv
import hydra
from omegaconf import DictConfig
# load environment variables from `.env` file if it exists
# recursively searches for `.env` in all folders starting from work dir
dotenv.load_dotenv(override=True)
@hydra.main(version_base = None, config_path="configs/", config_name="test.yaml")
def main(config: DictConfig):
# Imports can be nested inside @hydra.main to optimize tab completion
# https://github.com/facebookresearch/hydra/issues/934
from src import utils
from src.testing_pipeline import test
# Applies optional utilities
utils.extras(config)
# Evaluate model
return test(config)
if __name__ == "__main__":
main()