From 6b69b5eacc51f65acc80a3df6f457b60430b6764 Mon Sep 17 00:00:00 2001 From: hamid mohammadi Date: Sun, 8 Jan 2023 09:21:10 +0330 Subject: [PATCH] Fixed list type casting bug (#28) * Fixed list type casting bug * Fromatter applied to the trainer --- video_transformers/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/video_transformers/trainer.py b/video_transformers/trainer.py index e78bc30..780533b 100644 --- a/video_transformers/trainer.py +++ b/video_transformers/trainer.py @@ -1,5 +1,5 @@ from pathlib import Path -from typing import Dict, Union +from typing import Dict, List, Union import numpy as np import torch @@ -30,7 +30,7 @@ def __init__( mixed_precision: str = "no", output_dir: str = "runs", seed: int = 42, - trackers: list[GeneralTracker] = None, + trackers: List[GeneralTracker] = None, checkpoint_save: bool = True, checkpoint_save_interval: int = 1, checkpoint_save_policy: str = "epoch",