Skip to content

Commit

Permalink
Update documentation and ordering
Browse files Browse the repository at this point in the history
  • Loading branch information
JMGaljaard committed Sep 18, 2022
1 parent 0d23cb7 commit bfea85c
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 14 deletions.
3 changes: 3 additions & 0 deletions fltk/util/cluster/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,9 @@ class ClusterManager(abc.ABC):
Object with basic monitoring functionality. This shows how the information of different Pods in a cluster can be
requested and parsed. Currently, it mainly exists to start the ResourceWatchDog, which now only keeps track of the
amount of resources requested/used in the cluster.
This object is provided to the Orchestrator to allow for keeping track of available resources to allow for
scheduling of training tasks for extending an Orchestrator for experiments..
"""

def __init__(self):
Expand Down
25 changes: 12 additions & 13 deletions fltk/util/config/definitions/net.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,26 +4,25 @@

@unique
class Nets(Enum):
""" """
cifar100_resnet = "Cifar100ResNet"
cifar100_vgg = "Cifar100VGG"
cifar10_cnn = "Cifar10CNN"
cifar10_resnet = "Cifar10ResNet"
fashion_mnist_cnn = "FashionMNISTCNN"
fashion_mnist_resnet = "FashionMNISTResNet"
cifar100_resnet = 'Cifar100ResNet'
cifar100_vgg = 'Cifar100VGG'
cifar10_cnn = 'Cifar10CNN'
cifar10_resnet = 'Cifar10ResNet'
fashion_mnist_cnn = 'FashionMNISTCNN'
fashion_mnist_resnet = 'FashionMNISTResNet'
mnist_cnn = 'MNISTCNN'

@classmethod
def _missing_name_(cls, name: str) -> Optional["Nets"]:
"""Helper function to get lower/higher-case configured network, to allow for case-insensitive lookup.
def _missing_name_(cls, name: str) -> 'Dataset':
'''Helper function in case name could not be looked up (to support older configurations).
Args:
name (str): Name of network to lookup in case of missing lookup.
name (str): Name of Type to be looked up.
Returns:
Optional[Nets]: Of name is not part of defined networks, else reference to class implementing network.
Dataset: Corresponding Enum instance, if name is recognized from lower case.
"""
'''
for member in cls:
if member.name.lower() == name.lower():
return member
return member
2 changes: 1 addition & 1 deletion fltk/util/task/arrival_task.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ class ArrivalTask(abc.ABC):
loss_function: str = field(compare=False)
seed: int = field(compare=False)
replication: int = field(compare=False)
type_map: "Optional[FrozenOrderedDict[str, int]]"
type_map: "Optional[FrozenOrderedDict[str, int]]" = field(compare=False)
system_parameters: SystemParameters = field(compare=False)
hyper_parameters: HyperParameters = field(compare=False)
learning_parameters: LearningParameters = field(compare=False)
Expand Down

0 comments on commit bfea85c

Please sign in to comment.