ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 10:51:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

add: note

Browse Source
This commit is contained in:
Francesco Capuano
2025-09-27 12:13:53 +02:00
parent 5b647e3bcb
commit 939cb4d938
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lerobot/policies/sac/reward_model/modeling_classifier.py
View File

@@ -59,7 +59,9 @@ class SpatialLearnedEmbeddings(nn.Module):
super().__init__()
self.height = height
self.width = width
self.channel = channel
self.channel = (
channel # TODO(fracapuano): this gives issues with non-square images bc is hardcoded to 4
)
self.num_features = num_features
self.kernel = nn.Parameter(torch.empty(channel, height, width, num_features))