定义一个字典类:Dictclass。完成下面的功能: 1.del_dict(key):删除某个key, 如果没有key则返回“not found” 2.has_dict(key):判断某个键是否在字典里,如果在返回键对应的值,不存在则返回"not found" 3.get_keys():返回键组成的列表,返回类型为tuple 4.update_dict(dict2):将该对象与字典dict2的values合并,返回全部values组成的列表,返回类型为list 测试如下: dict1=Dictclass({'A':1,'B':0.1,'C','C'}) dict1.del_dict('A') dict1.del_dict('a') dict1.has_dict('B') dict1.has_dict('b') dict1.get_keys() dict1.update_dict({'D':5,'E':10}) 代码中不用try,excrpt
时间: 2023-06-12 09:02:10 浏览: 81
class Dictclass:
def __init__(self, my_dict):
self.my_dict = my_dict
def del_dict(self, key):
if key in self.my_dict:
del self.my_dict[key]
else:
return "not found"
def has_dict(self, key):
if key in self.my_dict:
return self.my_dict[key]
else:
return "not found"
def get_keys(self):
return tuple(self.my_dict.keys())
def update_dict(self, dict2):
self.my_dict.update(dict2)
return list(self.my_dict.values())
相关问题
del weights_dict[k] KeyError: 'pre_logits.fc.weight'
这个错误可能是因为你试图删除一个字典中不存在的键。请确保你的 `weights_dict` 字典中确实包含一个名为 `pre_logits.fc.weight` 的键。你可以使用 `if k in weights_dict:` 来检查一下是否存在该键,然后再尝试删除它。另外,也可能是因为你的字典是只读的,无法进行删除操作。如果是这种情况,可以尝试将字典复制一份,然后在复制的字典上进行删除操作。
给下列代码加注释: def merge_accumulate_client_update(self, list_num_proc, list_state_dict, lr): total_num_proc = sum(list_num_proc) # merged_state_dict = dict() dict_keys = list_state_dict[0].keys() for state_dict in list_state_dict[1:]: assert state_dict.keys() == dict_keys # accumulate extra sgrad and remove from state_dict if self.use_adaptive and self.is_adj_round(): prefix = "extra." for state_dict in list_state_dict: del_list = [] for key, param in state_dict.items(): if key[:len(prefix)] == prefix: sgrad_key = key[len(prefix):] mask_0 = self.model.get_mask_by_name(sgrad_key) == 0. dense_sgrad = torch.zeros_like(mask_0, dtype=torch.float) dense_sgrad.masked_scatter_(mask_0, param) # no need to divide by lr self.control.accumulate(sgrad_key, dense_sgrad) del_list.append(key) for del_key in del_list: del state_dict[del_key]
```python
def merge_accumulate_client_update(self, list_num_proc, list_state_dict, lr):
total_num_proc = sum(list_num_proc)
# merged_state_dict = dict()
dict_keys = list_state_dict[0].keys()
# Check if all state dicts have the same keys
for state_dict in list_state_dict[1:]:
assert state_dict.keys() == dict_keys
# accumulate extra sgrad and remove from state_dict
if self.use_adaptive and self.is_adj_round():
prefix = "extra."
for state_dict in list_state_dict:
del_list = []
for key, param in state_dict.items():
# Check if the key starts with 'extra.'
if key[:len(prefix)] == prefix:
# Get the corresponding sgrad key
sgrad_key = key[len(prefix):]
# Create a mask of zeroes
mask_0 = self.model.get_mask_by_name(sgrad_key) == 0.
# Create a dense tensor and fill it with values from param based on the mask
dense_sgrad = torch.zeros_like(mask_0, dtype=torch.float)
dense_sgrad.masked_scatter_(mask_0, param)
# Accumulate the dense sgrad without dividing by lr
self.control.accumulate(sgrad_key, dense_sgrad)
# Add the key to the delete list
del_list.append(key)
# Remove the keys from the state_dict
for del_key in del_list:
del state_dict[del_key]
```
这段代码实现了一个`merge_accumulate_client_update`方法,主要功能是合并和累加`list_state_dict`中的状态字典。以下是对代码的注释:
- `total_num_proc`:所有进程数的总和。
- `dict_keys`:状态字典的键列表。
- 检查所有状态字典是否具有相同的键。
- 如果使用自适应且处于调整轮次,则累加额外的`sgrad`并从状态字典中删除。
- `prefix`:额外`sgrad`的前缀。
- 对于每个状态字典,遍历键和参数。
- 如果键以`prefix`开头,则获取相应的`sgrad`键。
- 创建一个零填充的掩码。
- 创建一个稠密张量,并根据掩码从参数中填充值。
- 累加不除以`lr`的稠密`sgrad`。
- 将键添加到删除列表。
- 从状态字典中删除键。