From 028b1d5a374e54568003cc9d6e11a9153b030747 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Sun, 29 Oct 2023 19:24:05 +0000 Subject: [PATCH] revert trainer code --- nerfstudio/engine/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py index 56b147b35..6cfde69bc 100644 --- a/nerfstudio/engine/trainer.py +++ b/nerfstudio/engine/trainer.py @@ -24,7 +24,7 @@ import time from dataclasses import dataclass, field from pathlib import Path from threading import Lock -from typing import Dict, List, Literal, Optional, Tuple, Type, cast,Dict,DefaultDict +from typing import Dict, List, Literal, Optional, Tuple, Type, cast,Union,Dict,DefaultDict from collections import defaultdict import torch from nerfstudio.configs.experiment_config import ExperimentConfig