added model definitions
This commit is contained in:
@@ -0,0 +1,8 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from .fully_connected import FullyConnectedModel
|
||||
|
||||
|
||||
class ContactTail(FullyConnectedModel):
|
||||
def __init__(self):
|
||||
super().__init__(num_out_features=2)
|
||||
Reference in New Issue
Block a user