From 387a602e1a3ba0efc8a76c9c4e5e97e307823ef8 Mon Sep 17 00:00:00 2001 From: hamid mohammadi Date: Sat, 31 Dec 2022 12:39:03 +0330 Subject: [PATCH 1/2] Fixed list type casting bug --- 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..676a633 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, Union, List 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", From 02d370466db224c932b5f92c6e1efe4f79a704b2 Mon Sep 17 00:00:00 2001 From: sandstorm12 Date: Sat, 7 Jan 2023 18:38:09 +0330 Subject: [PATCH 2/2] Fromatter applied to the trainer --- video_transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/video_transformers/trainer.py b/video_transformers/trainer.py index 676a633..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, List +from typing import Dict, List, Union import numpy as np import torch