Commit 5dcd990e authored by Rafael Dätwyler's avatar Rafael Dätwyler
Browse files

rename 'actions' to 'classes' in controller

parent 6b8022df
......@@ -4,9 +4,9 @@ import torch.nn as nn
class Controller(nn.Module):
""" Controller """
def __init__(self, latents, recurrents, actions):
def __init__(self, latents, recurrents, classes):
super().__init__()
self.fc = nn.Linear(latents + recurrents, actions)
self.fc = nn.Linear(latents + recurrents, classes)
def forward(self, *inputs):
cat_in = torch.cat(inputs, dim=1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment