INSERT GRAPHIC HERE (include hyperlink in image)
Subtitle or Short Description Goes Here
ideally one sentence >
from torch.optim.sgd import SGD | |
from torch.optim.optimizer import required | |
class PGM(SGD): | |
def __init__(self, params, proxs, lr=required, momentum=0, dampening=0, | |
nesterov=False): | |
kwargs = dict(lr=lr, momentum=momentum, dampening=dampening, weight_decay=0, nesterov=nesterov) | |
super().__init__(params, **kwargs) | |
if len(proxs) != len(self.param_groups): | |
raise ValueError("Invalid length of argument proxs: {} instead of {}".format(len(proxs), len(self.param_groups))) |
#include <iostream> | |
using namespace std; | |
void print(){cout<<'\n';} | |
template<typename T, typename ...TAIL> | |
void print(const T &t, TAIL... tail) | |
{ | |
cout<<t<<' '; |