update: actpred 默认图像分辨率 64→224
This commit is contained in:
1 parent
fee1241e49
commit
6d79fc919f
4 files changed
+5
-5
No files matched your search
+1
-1
@@ -16,7 +16,7 @@ def _natural_key(p):
|
||||
|
||||
class ActionDataset(Dataset):
|
||||
def __init__(self, data_dir: str, history_len: int, action_dim: int,
|
||||
image_size: int = 64, normalize_actions: bool = False,
|
||||
image_size: int = 224, normalize_actions: bool = False,
|
||||
action_stats_path: str = None, episode_indices: list = None,
|
||||
frame_len: int = 3):
|
||||
self.history_len = history_len
|
||||
|
||||
Reference in new issue
Block a user