Skip to content

train.warn_if_small_image_observation

Full path: schola.scripts.sb3.train.warn_if_small_image_observation

schola.scripts.sb3.train.warn_if_small_image_observation

warn_if_small_image_observation

warn_if_small_image_observation(observation_space, threshold=64)

Issue a panel warning if any Box observation that looks image-like has a spatial dimension smaller than threshold.

Heuristic: Treat Box spaces with shape len 2 (H, W) or len 3 (C,H,W) / (H,W,C) as image-like. For 3D, pick the two largest dims as spatial. If min spatial < threshold we warn once.

Parameters

threshold (int)