mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-16 17:20:05 +00:00
fixup! fixup! fixup! Improve visualization: separate correction plot and fix axis scaling
This commit is contained in:
@@ -21,7 +21,7 @@ import torch
|
||||
from matplotlib.figure import Figure
|
||||
from torch import Tensor
|
||||
|
||||
from lerobot.policies.rtc.debug_handler import Tracker
|
||||
from lerobot.policies.rtc.debug_tracker import Tracker
|
||||
|
||||
|
||||
class RTCDebugVisualizer:
|
||||
|
||||
@@ -29,7 +29,7 @@ from torch import Tensor
|
||||
|
||||
from lerobot.configs.types import RTCAttentionSchedule
|
||||
from lerobot.policies.rtc.configuration_rtc import RTCConfig
|
||||
from lerobot.policies.rtc.debug_handler import Tracker
|
||||
from lerobot.policies.rtc.debug_tracker import Tracker
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user