def main():
#train_pwl = MNISTData.load_train().to_pwl().restrict_to(set(range(10))).rescale()
test_pwl = MNISTData.load_test().to_pwl().restrict_to(set(range(10))).rescale()
layers_and_nonlins = (
(90, 'tanh'),
(90, 'tanh'),
(90, 'tanh'),
)
#layers = [lyr[0] for lyr in layers_and_nonlins]
nonlins = [lyr[1] for lyr in layers_and_nonlins]
nonlins.append('linear') # output
#layer_names = [f'{lyr[1]} (layer {idx})' for idx, lyr in enumerate(layers_and_nonlins)]
layer_names = [f'Layer {idx+1}' for idx, lyr in enumerate(layers_and_nonlins)]
layer_names.insert(0, 'input')
layer_names.append('output')
network = torch.load(INPFILE)
dig = npmp.NPDigestor('train_one', 8)
pca3d_dir = os.path.join(SAVEDIR, 'pca3d')
traj = pca_ff.find_trajectory(network, test_pwl, 3)
pca_3d.plot_ff(traj, pca3d_dir, False, 16.67, dig, layer_names=layer_names)
dig.join()
def train(variances, reuse_repeats, num_repeats):
"""Trains all the networks"""
dig = npmp.NPDigestor(
'train_mult_contr_noise',
4,
target_module='mnist.runners.train_multiple_contrast_noise',
target_name='train_with_noise')
empty_arr = np.array([])
for vari in variances:
for i in range(reuse_repeats, num_repeats):
dig(vari, i, empty_arr)
dig.join()
def digest_measure_and_plot_pr_ff(sample_points: np.ndarray, sample_labels: np.ndarray,
output_dim: int,
*all_hid_acts: typing.Tuple[np.ndarray],
savepath: str = None, labels: bool = False,
max_threads: typing.Optional[int] = 3):
"""An npmp digestable callable for measuring and plotting the participation ratio for
a feedforward network"""
if not isinstance(output_dim, int):
raise ValueError(f'expected output_dim is int, got {output_dim} (type={type(output_dim)})')
sample_points = torch.from_numpy(sample_points)
sample_labels = torch.from_numpy(sample_labels)
hacts_cp = []
for hact in all_hid_acts:
hacts_cp.append(torch.from_numpy(hact))
all_hid_acts = hacts_cp
num_lyrs = len(all_hid_acts)
if labels:
masks_by_lbl = [sample_labels == lbl for lbl in range(output_dim)]
inqueue = myq.ZeroMQQueue.create_recieve()
inq_serd = inqueue.serd()
dig = npmp.NPDigestor(uuid.uuid4().hex, max_threads, 'shared.measures.participation_ratio', 'measure_pr_np')
exp_results = len(all_hid_acts)
if labels:
exp_results += len(all_hid_acts) * output_dim
for layer, hid_acts in enumerate(all_hid_acts):
dig(hid_acts.numpy(), (layer, -1), inq_serd)
if labels:
for lbl in range(output_dim):
dig(hid_acts[masks_by_lbl[lbl]].numpy(), (layer, lbl), inq_serd)
torch_pr_overall = torch.zeros(num_lyrs, dtype=torch.double)
if labels:
torch_pr_by_label = torch.zeros((output_dim, num_lyrs), dtype=torch.double)
for _ in range(exp_results):
(layer, lbl), prval = inqueue.get()
if lbl == -1:
torch_pr_overall[layer] = prval
else:
torch_pr_by_label[lbl, layer] = prval
traj = PRTrajectory(overall=torch_pr_overall,
by_label=torch_pr_by_label if labels else None,
layers=True)
plot_pr_trajectory(traj, savepath, False)
def main():
"""Entry point"""
pwl = GaussianSpheresPWLP(epoch_size=1000,
input_dim=2,
output_dim=2,
clusters=[
PointWithLabel(point=torch.tensor(
(-1, 0), dtype=torch.double),
label=0),
PointWithLabel(point=torch.tensor(
(1, 0), dtype=torch.double),
label=1)
],
std_dev=0.4,
mean=0)
layers = [(50, True, False)]
layer_names = ['input', 'hidden', 'output']
network = FeedforwardLarge.create(input_dim=2,
output_dim=2,
weights=wi.GaussianWeightInitializer(
mean=0, vari=0.1, normalize_dim=1),
biases=wi.ZerosWeightInitializer(),
layer_sizes=layers,
nonlinearity='linear',
train_readout_weights=False,
train_readout_bias=False)
trainer = tnr.GenericTrainer(
train_pwl=pwl,
test_pwl=pwl,
teacher=FFTeacher(),
batch_size=1,
learning_rate=0.003,
optimizer=torch.optim.Adam(
[p for p in network.parameters() if p.requires_grad], lr=0.003),
criterion=mycrits.create_meansqerr_regul(
noise_strength=0.5) #torch.nn.CrossEntropyLoss()
)
pca3d_throughtrain.FRAMES_PER_TRAIN = 1
pca3d_throughtrain.SKIP_TRAINS = 0
pca3d_throughtrain.NUM_FRAME_WORKERS = 4
dig = npmp.NPDigestor('train_one', 5)
#pca_3d.plot_ff(pca_ff.find_trajectory(network, pwl, 3), os.path.join(SAVEDIR, 'pca_3d_start'), True,
# digestor=dig, frame_time=FRAME_TIME, layer_names=layer_names)
dtt_training_dir = os.path.join(SAVEDIR, 'dtt')
pca_training_dir = os.path.join(SAVEDIR, 'pca')
pr_training_dir = os.path.join(SAVEDIR, 'pr')
svm_training_dir = os.path.join(SAVEDIR, 'svm')
satur_training_dir = os.path.join(SAVEDIR, 'saturation')
pca_throughtrain_dir = os.path.join(SAVEDIR, 'pca_throughtrain')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(10)).reg(
tnr.DecayTracker())
#.reg(tnr.DecayStopper(8))
#.reg(tnr.LRMultiplicativeDecayer())
.reg(tnr.DecayOnPlateau()).reg(tnr.AccuracyTracker(5, 1000, True))
#.reg(tnr.WeightNoiser(
# wi.GaussianWeightInitializer(mean=0, vari=0.02, normalize_dim=None),
# lambda ctxt: ctxt.model.layers[-1].weight.data))
.reg(
tnr.OnEpochCaller.create_every(
satur.during_training(satur_training_dir, True, dig),
skip=10)).reg(
tnr.OnEpochCaller.create_every(
dtt.during_training_ff(dtt_training_dir, True, dig),
skip=10)).reg(
tnr.OnEpochCaller.create_every(
pca_ff.during_training(pca_training_dir,
True,
dig,
alpha=0.8),
skip=10)).reg(
tnr.OnEpochCaller.create_every(
pr.during_training_ff(
pr_training_dir, True, dig),
skip=1000)).reg(
tnr.OnEpochCaller.create_every(
svm.during_training_ff(
svm_training_dir, True, dig),
skip=1000))
#.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join())).reg(
tnr.ZipDirOnFinish(dtt_training_dir)).reg(
tnr.ZipDirOnFinish(pca_training_dir)).reg(
tnr.ZipDirOnFinish(pr_training_dir)).reg(
tnr.ZipDirOnFinish(svm_training_dir)).reg(
tnr.ZipDirOnFinish(satur_training_dir)))
trainer.train(network)
#pca_3d.plot_ff(pca_ff.find_trajectory(network, pwl, 3), os.path.join(SAVEDIR, 'pca_3d_end'), True,
# digestor=dig, frame_time=FRAME_TIME, layer_names=layer_names)
dig.archive_raw_inputs(os.path.join(SAVEDIR, 'raw_digestor.zip'))
def train_with_noise(vari, rep, ignoreme): # pylint: disable=unused-argument
"""Entry point"""
train_pwl = MNISTData.load_train().to_pwl().restrict_to(set(
range(10))).rescale()
test_pwl = MNISTData.load_test().to_pwl().restrict_to(set(
range(10))).rescale()
layers_and_nonlins = (
(90, 'tanh'),
(90, 'tanh'),
(90, 'tanh'),
(90, 'tanh'),
(90, 'tanh'),
)
layers = [lyr[0] for lyr in layers_and_nonlins]
nonlins = [lyr[1] for lyr in layers_and_nonlins]
nonlins.append('tanh') # output
#layer_names = [f'{lyr[1]} (layer {idx})' for idx, lyr in enumerate(layers_and_nonlins)]
layer_names = [
f'Layer {idx+1}' for idx, lyr in enumerate(layers_and_nonlins)
]
layer_names.insert(0, 'Input')
layer_names.append('Output')
network = FeedforwardLarge.create(input_dim=train_pwl.input_dim,
output_dim=train_pwl.output_dim,
weights=wi.GaussianWeightInitializer(
mean=0, vari=0.3, normalize_dim=0),
biases=wi.ZerosWeightInitializer(),
layer_sizes=layers,
nonlinearity=nonlins
#layer_sizes=[500, 200]
)
_lr = 0.1
trainer = tnr.GenericTrainer(
train_pwl=train_pwl,
test_pwl=test_pwl,
teacher=FFTeacher(),
batch_size=30,
learning_rate=_lr,
optimizer=torch.optim.SGD(
[p for p in network.parameters() if p.requires_grad], lr=_lr
), #torch.optim.Adam([p for p in network.parameters() if p.requires_grad], lr=0.003),
criterion=mycrits.meansqerr #torch.nn.CrossEntropyLoss()#
)
#pca3d_throughtrain.FRAMES_PER_TRAIN = 4
#pca3d_throughtrain.SKIP_TRAINS = 0
#pca3d_throughtrain.NUM_FRAME_WORKERS = 6
dig = npmp.NPDigestor(f'TRMCN_{rep}_{vari}', 8)
savedir = os.path.join(SAVEDIR, f'variance_{vari}', f'repeat_{rep}')
dtt_training_dir = os.path.join(savedir, 'dtt')
pca_training_dir = os.path.join(savedir, 'pca')
pca3d_training_dir = os.path.join(savedir, 'pca3d')
pr_training_dir = os.path.join(savedir, 'pr')
svm_training_dir = os.path.join(savedir, 'svm')
satur_training_dir = os.path.join(savedir, 'saturation')
trained_net_dir = os.path.join(savedir, 'trained_model')
pca_throughtrain_dir = os.path.join(savedir, 'pca_throughtrain')
logpath = os.path.join(savedir, 'log.txt')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(0.2)).reg(
tnr.DecayTracker()).reg(tnr.DecayStopper(5)).reg(
tnr.LRMultiplicativeDecayer())
#.reg(tnr.DecayOnPlateau())
#.reg(tnr.DecayEvery(5))
.reg(tnr.AccuracyTracker(1, 1000, True)).reg(
tnr.WeightNoiser(
wi.GaussianWeightInitializer(mean=0, vari=vari),
(lambda ctx: ctx.model.layers[-1].weight.data.detach()), 'scale',
(lambda noise: wi.GaussianWeightInitializer(0, noise.vari * 0.5)
)))
#.reg(tnr.OnEpochCaller.create_every(dtt.during_training_ff(dtt_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_3d.during_training(pca3d_training_dir, True, dig, plot_kwargs={'layer_names': layer_names}), start=500, skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_ff.during_training(pca_training_dir, True, dig), skip=100))
.reg(
tnr.OnEpochCaller.create_every(pr.during_training_ff(
pr_training_dir, True, dig),
skip=1))
#.reg(tnr.OnEpochCaller.create_every(svm.during_training_ff(svm_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(satur.during_training(satur_training_dir, True, dig), skip=100))
.reg(
tnr.OnEpochCaller.create_every(tnr.save_model(trained_net_dir),
skip=100))
#.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join())).reg(
tnr.CopyLogOnFinish(logpath)).reg(
tnr.ZipDirOnFinish(dtt_training_dir)).reg(
tnr.ZipDirOnFinish(pca_training_dir)).reg(
tnr.ZipDirOnFinish(pca3d_training_dir)).reg(
tnr.ZipDirOnFinish(pr_training_dir)).reg(
tnr.ZipDirOnFinish(svm_training_dir)).reg(
tnr.ZipDirOnFinish(satur_training_dir)).reg(
tnr.ZipDirOnFinish(trained_net_dir)))
trainer.train(network)
dig.archive_raw_inputs(os.path.join(savedir, 'digestor_raw.zip'))
def main():
"""Entry point"""
pwl = GaussianSpheresPWLP.create(epoch_size=2700,
input_dim=INPUT_DIM,
output_dim=OUTPUT_DIM,
cube_half_side_len=2,
num_clusters=10,
std_dev=0.04,
mean=0,
min_sep=0.1)
nets = cu.FluentShape(INPUT_DIM).verbose()
network = FeedforwardComplex(INPUT_DIM, OUTPUT_DIM, [
nets.linear_(90),
nets.nonlin('isrlu'),
nets.linear_(OUTPUT_DIM),
])
trainer = tnr.GenericTrainer(
train_pwl=pwl,
test_pwl=pwl,
teacher=FFTeacher(),
batch_size=45,
learning_rate=0.001,
optimizer=torch.optim.Adam(
[p for p in network.parameters() if p.requires_grad], lr=0.001),
criterion=torch.nn.CrossEntropyLoss())
dig = npmp.NPDigestor('train_one_complex', 16)
#pca_3d.plot_ff(pca_ff.find_trajectory(network, pwl, 3), os.path.join(SAVEDIR, 'pca_3d_start'), True, dig3d)
#dig3d.join()
#exit()
dtt_training_dir = os.path.join(SAVEDIR, 'dtt')
pca_training_dir = os.path.join(SAVEDIR, 'pca')
pr_training_dir = os.path.join(SAVEDIR, 'pr')
svm_training_dir = os.path.join(SAVEDIR, 'svm')
satur_training_dir = os.path.join(SAVEDIR, 'saturation')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(150)).reg(
tnr.DecayTracker()).reg(tnr.DecayStopper(3)).reg(
tnr.LRMultiplicativeDecayer()).reg(tnr.DecayOnPlateau()).reg(
tnr.AccuracyTracker(5, 1000, True)).reg(
tnr.OnEpochCaller.create_every(dtt.during_training_ff(
dtt_training_dir, True),
skip=1000))
#.reg(tnr.OnEpochCaller.create_every(pca_ff.during_training(pca_training_dir, True), skip=1000))
.reg(
tnr.OnEpochCaller.create_every(
pr.during_training_ff(pr_training_dir, True), skip=1000)).reg(
tnr.OnEpochCaller.create_every(
svm.during_training_ff(svm_training_dir, True),
skip=1000)).reg(
tnr.OnEpochCaller.create_every(satur.during_training(
satur_training_dir, True),
skip=1000)).
reg(tnr.ZipDirOnFinish(dtt_training_dir)).reg(
tnr.ZipDirOnFinish(pca_training_dir)).reg(
tnr.ZipDirOnFinish(pr_training_dir)).reg(
tnr.ZipDirOnFinish(svm_training_dir)).reg(
tnr.ZipDirOnFinish(satur_training_dir)))
trainer.train(network)
torch.save(network.state_dict(), os.path.join(SAVEDIR,
'trained_network.pt'))
def main():
"""Entry point"""
pwl = GaussianSpheresPWLP.create(epoch_size=2700,
input_dim=INPUT_DIM,
output_dim=OUTPUT_DIM,
cube_half_side_len=2,
num_clusters=10,
std_dev=0.5,
mean=0,
min_sep=1,
force_split=True)
layers_and_nonlins = (
(100, 'tanh'),
#(100, 'linear'),
#(25, 'linear'),
#(90, 'tanh'),
#(90, 'tanh'),
#(90, 'linear'),
#(25, 'linear'),
)
layers = [lyr[0] for lyr in layers_and_nonlins]
nonlins = [lyr[1] for lyr in layers_and_nonlins]
nonlins.append('tanh') # output
layer_names = [
f'{lyr[1]} ({idx})' for idx, lyr in enumerate(layers_and_nonlins)
]
layer_names.insert(0, 'input')
layer_names.append('output')
network = FeedforwardLarge.create(input_dim=INPUT_DIM,
output_dim=OUTPUT_DIM,
weights=wi.GaussianWeightInitializer(
mean=0, vari=0.3, normalize_dim=1),
biases=wi.ZerosWeightInitializer(),
layer_sizes=layers,
nonlinearity=nonlins)
trainer = tnr.GenericTrainer(
train_pwl=pwl,
test_pwl=pwl,
teacher=FFTeacher(),
batch_size=20,
learning_rate=0.001,
optimizer=torch.optim.Adam(
[p for p in network.parameters() if p.requires_grad], lr=0.001),
criterion=mycrits.meansqerr #torch.nn.CrossEntropyLoss()
)
pca3d_throughtrain.FRAMES_PER_TRAIN = 1
pca3d_throughtrain.SKIP_TRAINS = 4
pca3d_throughtrain.NUM_FRAME_WORKERS = 6
dig = npmp.NPDigestor('train_one', 35)
#pca_3d.plot_ff(pca_ff.find_trajectory(network, pwl, 3), os.path.join(SAVEDIR, 'pca_3d_start'), True,
# digestor=dig, frame_time=FRAME_TIME, layer_names=layer_names)
dtt_training_dir = os.path.join(SAVEDIR, 'dtt')
pca_training_dir = os.path.join(SAVEDIR, 'pca')
pr_training_dir = os.path.join(SAVEDIR, 'pr')
svm_training_dir = os.path.join(SAVEDIR, 'svm')
satur_training_dir = os.path.join(SAVEDIR, 'saturation')
pca_throughtrain_dir = os.path.join(SAVEDIR, 'pca_throughtrain')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(100)).reg(
tnr.InfOrNANDetecter()).reg(tnr.DecayTracker()).reg(
tnr.DecayStopper(8)).reg(tnr.LRMultiplicativeDecayer()).reg(
tnr.DecayOnPlateau()).reg(tnr.AccuracyTracker(5, 1000, True))
#.reg(tnr.WeightNoiser(
# wi.GaussianWeightInitializer(mean=0, vari=0.02, normalize_dim=None),
# lambda ctxt: ctxt.model.layers[-1].weight.data))
#.reg(tnr.OnEpochCaller.create_every(satur.during_training(satur_training_dir, True, dig), skip=1000))
#.reg(tnr.OnEpochCaller.create_every(dtt.during_training_ff(dtt_training_dir, True, dig), skip=1000))
.reg(
tnr.OnEpochCaller.create_every(pca_ff.during_training(
pca_training_dir, True, dig),
skip=1000))
#.reg(tnr.OnEpochCaller.create_every(pr.during_training_ff(pr_training_dir, True, dig), skip=1000))
#.reg(tnr.OnEpochCaller.create_every(svm.during_training_ff(svm_training_dir, True, dig), skip=1000))
#.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join())).reg(
tnr.ZipDirOnFinish(dtt_training_dir)).reg(
tnr.ZipDirOnFinish(pca_training_dir)).reg(
tnr.ZipDirOnFinish(pr_training_dir)).reg(
tnr.ZipDirOnFinish(svm_training_dir)).reg(
tnr.ZipDirOnFinish(satur_training_dir)))
trainer.train(network)
#pca_3d.plot_ff(pca_ff.find_trajectory(network, pwl, 3), os.path.join(SAVEDIR, 'pca_3d_end'), True,
# digestor=dig, frame_time=FRAME_TIME, layer_names=layer_names)
dig.archive_raw_inputs(os.path.join(SAVEDIR, 'raw_digestor.zip'))
def main():
"""Entry point"""
train_pwl = MNISTData.load_train().to_pwl().restrict_to(set(range(10))).rescale()
test_pwl = MNISTData.load_test().to_pwl().restrict_to(set(range(10))).rescale()
network = NaturalRNN.create(
'tanh', train_pwl.input_dim, 200, train_pwl.output_dim,
input_weights=wi.OrthogonalWeightInitializer(0.03, 0),
input_biases=wi.ZerosWeightInitializer(), #
hidden_weights=wi.SompolinskySmoothedFixedGainWeightInitializer(0.001, 20),
hidden_biases=wi.GaussianWeightInitializer(mean=0, vari=0.3, normalize_dim=0),
output_weights=wi.GaussianWeightInitializer(mean=0, vari=0.3, normalize_dim=0),
output_biases=wi.ZerosWeightInitializer()
)
trainer = tnr.GenericTrainer(
train_pwl=train_pwl,
test_pwl=test_pwl,
teacher=RNNTeacher(recurrent_times=10, input_times=1),
batch_size=30,
learning_rate=0.0001,
optimizer=torch.optim.RMSprop([p for p in network.parameters() if p.requires_grad], lr=0.0001, alpha=0.9),
criterion=torch.nn.CrossEntropyLoss()
)
(trainer
.reg(tnr.EpochsTracker())
.reg(tnr.EpochsStopper(150))
.reg(tnr.InfOrNANDetecter())
.reg(tnr.InfOrNANDetecter())
.reg(tnr.DecayTracker())
.reg(tnr.DecayStopper(5))
.reg(tnr.LRMultiplicativeDecayer())
.reg(tnr.DecayOnPlateau())
.reg(tnr.AccuracyTracker(5, 1000, True))
)
print('--saving pcs before training--')
traj = pca.find_trajectory(network, train_pwl, 10, 2)
savepath = os.path.join(SAVEDIR, 'pca_before_train')
pca.plot_trajectory(traj, savepath, exist_ok=True)
traj = pca.find_trajectory(network, test_pwl, 10, 2)
savepath = os.path.join(SAVEDIR, 'pca_before_test')
pca.plot_trajectory(traj, savepath, exist_ok=True)
del traj
# print('--saving distance through time before training--')
# savepath = os.path.join(SAVEDIR, 'dtt_before_train')
# dtt.measure_dtt(network, train_pwl, 10, savepath, verbose=True, exist_ok=True)
# savepath = os.path.join(SAVEDIR, 'dtt_before_test')
# dtt.measure_dtt(network, test_pwl, 10, savepath, verbose=True, exist_ok=True)
print('--training--')
result = trainer.train(network)
print('--finished training--')
print(result)
print('--saving pcs after training--')
traj = pca.find_trajectory(network, train_pwl, 10, 2)
savepath = os.path.join(SAVEDIR, 'pca_after_train')
pca.plot_trajectory(traj, savepath, exist_ok=True)
traj = pca.find_trajectory(network, test_pwl, 10, 2)
savepath = os.path.join(SAVEDIR, 'pca_after_test')
pca.plot_trajectory(traj, savepath, exist_ok=True)
del traj
# print('--saving distance through time after training--')
# savepath = os.path.join(SAVEDIR, 'dtt_after_train')
# dtt.measure_dtt(network, train_pwl, 10, savepath, verbose=True, exist_ok=True)
# savepath = os.path.join(SAVEDIR, 'dtt_after_test')
# dtt.measure_dtt(network, test_pwl, 10, savepath, verbose=True, exist_ok=True)
print('--saving 3d pca plots after training--')
layer_names = ['Input']
for i in range(1, trainer.teacher.recurrent_times + 1):
layer_names.append(f'Timestep {i}')
dig = npmp.NPDigestor('mnist_train_one_rnn', 2)
nha = mutils.get_hidacts_rnn(network, train_pwl, trainer.teacher.recurrent_times)
nha.torch()
traj = pca_ff.to_trajectory(nha.sample_labels, nha.hid_acts, 3)
pca_3d.plot_ff(traj, os.path.join(SAVEDIR, 'pca3d_after_train'), False, digestor=dig,
layer_names=layer_names)
nha = mutils.get_hidacts_rnn(network, test_pwl, trainer.teacher.recurrent_times)
nha.torch()
traj = pca_ff.to_trajectory(nha.sample_labels, nha.hid_acts, 3)
pca_3d.plot_ff(traj, os.path.join(SAVEDIR, 'pca3d_after_test'), False, digestor=dig,
layer_names=layer_names)
print('--saving model--')
torch.save(network, os.path.join(SAVEDIR, 'model.pt'))
dig.join()
def train_with_noise(vari, rep, pr_repeats, ignoreme): # pylint: disable=unused-argument
"""Entry point"""
train_pwl = GaussianSpheresPWLP.create(epoch_size=30000,
input_dim=INPUT_DIM,
output_dim=2,
cube_half_side_len=2,
num_clusters=10,
std_dev=0.2,
mean=0,
min_sep=0.4,
force_split=True)
test_pwl = train_pwl
nets = cu.FluentShape(INPUT_DIM).verbose()
mywi = wi.WICombine([
wi.RectangularEyeWeightInitializer(1),
wi.GaussianWeightInitializer(mean=0, vari=0.3)
])
network = FeedforwardComplex(INPUT_DIM, train_pwl.output_dim, [
nets.linear_(DIM, weights_init=mywi),
nets.nonlin('leakyrelu'),
nets.linear_(DIM, weights_init=mywi),
nets.nonlin('leakyrelu'),
nets.linear_(DIM, weights_init=mywi),
nets.nonlin('leakyrelu'),
nets.linear_(DIM, weights_init=mywi),
nets.nonlin('leakyrelu'),
nets.linear_(train_pwl.output_dim),
nets.nonlin('leakyrelu'),
])
_lr = 0.01
trainer = tnr.GenericTrainer(
train_pwl=train_pwl,
test_pwl=test_pwl,
teacher=FFTeacher(),
batch_size=20,
learning_rate=_lr,
optimizer=torch.optim.SGD(
[p for p in network.parameters() if p.requires_grad], lr=_lr),
criterion=mycrits.hubererr #torch.nn.CrossEntropyLoss()#
)
#pca3d_throughtrain.FRAMES_PER_TRAIN = 4
#pca3d_throughtrain.SKIP_TRAINS = 0
#pca3d_throughtrain.NUM_FRAME_WORKERS = 6
dig = npmp.NPDigestor(f'TRMCN_{rep}_{vari}', 4)
savedir = os.path.join(SAVEDIR, f'variance_{vari}', f'repeat_{rep}')
shared.filetools.deldir(savedir)
dtt_training_dir = os.path.join(savedir, 'dtt')
pca_training_dir = os.path.join(savedir, 'pca')
pca3d_training_dir = os.path.join(savedir, 'pca3d')
pr_training_dir = os.path.join(savedir, 'pr')
svm_training_dir = os.path.join(savedir, 'svm')
satur_training_dir = os.path.join(savedir, 'saturation')
trained_net_dir = os.path.join(savedir, 'trained_model')
pca_throughtrain_dir = os.path.join(savedir, 'pca_throughtrain')
acts_training_dir = os.path.join(savedir, 'acts')
logpath = os.path.join(savedir, 'log.txt')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(300)).reg(
tnr.EpochProgress(5, hint_end_epoch=10)).reg(tnr.DecayTracker()).reg(
tnr.DecayStopper(10)).reg(tnr.InfOrNANDetecter()).reg(
tnr.InfOrNANStopper()).reg(
tnr.LRMultiplicativeDecayer(factor=0.9))
#.reg(tnr.DecayOnPlateau(verbose=False))
.reg(tnr.DecayEvery(1, verbose=False)).reg(
tnr.AccuracyTracker(1,
1000,
True,
savepath=os.path.join(savedir, 'accuracy.json'))))
if ALL_LAYERS_NOISED:
tonoise = list(range(1, len(network.layers)))
else:
tonoise = [len(network.layers) - 2]
noisestyle = 'add'
def layer_fetcher(lyr):
return lambda ctx: ctx.model.layers[lyr].action.weight.data.detach()
noisedecayer = lambda noise: wi.GaussianWeightInitializer(
0, noise.vari * 0.9)
for lyr in tonoise:
if network.layers[lyr].is_module:
trainer.reg(
tnr.WeightNoiser(
wi.GaussianWeightInitializer(mean=0, vari=vari),
layer_fetcher(lyr), noisestyle, noisedecayer))
if rep < pr_repeats:
trainer.reg(
tnr.OnEpochCaller.create_every(pr.during_training_ff(
pr_training_dir, True, dig),
skip=100))
(trainer
#.reg(tnr.OnEpochCaller.create_every(dtt.during_training_ff(dtt_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_3d.during_training(pca3d_training_dir, True, dig, plot_kwargs={'layer_names': layer_names}), start=500, skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_ff.during_training(pca_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(pr.during_training_ff(pr_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(svm.during_training_ff(svm_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(satur.during_training(satur_training_dir, True, dig), skip=100))
.reg(tnr.OnEpochCaller.create_every(measacts.during_training(acts_training_dir, dig, meta={'time': time.time(), 'noised_layers': tonoise, 'variance': vari, 'repeat': rep}), skip=100))
.reg(tnr.OnEpochCaller.create_every(tnr.save_model(trained_net_dir), skip=100))
#.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join()))
.reg(tnr.CopyLogOnFinish(logpath))
.reg(tnr.ZipDirOnFinish(dtt_training_dir))
.reg(tnr.ZipDirOnFinish(pca_training_dir))
.reg(tnr.ZipDirOnFinish(pca3d_training_dir))
.reg(tnr.ZipDirOnFinish(pr_training_dir))
.reg(tnr.ZipDirOnFinish(svm_training_dir))
.reg(tnr.ZipDirOnFinish(satur_training_dir))
.reg(tnr.ZipDirOnFinish(trained_net_dir))
)
result = trainer.train(network)
dig.archive_raw_inputs(os.path.join(savedir, 'digestor_raw.zip'))
if result['inf_or_nan']:
print('[TMCN] Inf or NAN detected - repeating run')
shared.filetools.deldir(savedir)
def main():
"""Entry point"""
nets = cu.FluentShape(28 * 28).verbose()
network = FeedforwardComplex(INPUT_DIM, OUTPUT_DIM, [
nets.linear_(HIDDEN_DIM),
nets.tanh(),
nets.linear_(OUTPUT_DIM),
nets.tanh()
])
train_pwl = MNISTData.load_train().to_pwl().restrict_to(set(
range(10))).rescale()
test_pwl = MNISTData.load_test().to_pwl().restrict_to(set(
range(10))).rescale()
layer_names = ('Input', 'Hidden', 'Output')
trainer = tnr.GenericTrainer(
train_pwl=train_pwl,
test_pwl=test_pwl,
teacher=FFTeacher(),
batch_size=45,
learning_rate=0.001,
optimizer=torch.optim.Adam(
[p for p in network.parameters() if p.requires_grad], lr=0.001),
criterion=mycrits.meansqerr #torch.nn.CrossEntropyLoss()
)
dig = npmp.NPDigestor('train_one_complex', 35)
dtt_training_dir = os.path.join(SAVEDIR, 'dtt')
pca_training_dir = os.path.join(SAVEDIR, 'pca')
pca3d_training_dir = os.path.join(SAVEDIR, 'pca3d')
pr_training_dir = os.path.join(SAVEDIR, 'pr')
svm_training_dir = os.path.join(SAVEDIR, 'svm')
satur_training_dir = os.path.join(SAVEDIR, 'saturation')
trained_net_dir = os.path.join(SAVEDIR, 'trained_model')
pca_throughtrain_dir = os.path.join(SAVEDIR, 'pca_throughtrain')
wds_training_dir = os.path.join(SAVEDIR, 'weightdeltas')
logpath = os.path.join(SAVEDIR, 'log.txt')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(3)).reg(
tnr.DecayTracker()).reg(tnr.DecayStopper(8)).reg(
tnr.LRMultiplicativeDecayer()).reg(tnr.DecayOnPlateau()).
reg(tnr.AccuracyTracker(5, 1000, True)).reg(
tnr.WeightNoiser(
wi.GaussianWeightInitializer(mean=0,
vari=0.1),
(lambda ctx: ctx.model.layers[0].action.weight.data.detach()),
'scale',
(lambda noise: wi.GaussianWeightInitializer(0, noise.vari * 0.5)
))).reg(
tnr.OnEpochCaller.create_every(dtt.during_training_ff(
dtt_training_dir, True, dig),
skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_3d.during_training(pca3d_training_dir, True, dig, plot_kwargs={'layer_names': layer_names}), start=1000, skip=1000))
.reg(
tnr.OnEpochCaller.create_every(
pca_ff.during_training(pca_training_dir, True, dig),
skip=100)).reg(
tnr.OnEpochCaller.create_every(
pr.during_training_ff(pr_training_dir, True, dig),
skip=100)).reg(
tnr.OnEpochCaller.create_every(
svm.during_training_ff(svm_training_dir, True,
dig),
skip=100)).reg(
tnr.OnEpochCaller.create_every(
satur.during_training(
satur_training_dir, True, dig),
skip=100)).reg(
tnr.OnEpochCaller.create_every(
tnr.save_model(trained_net_dir),
skip=100)).
reg(
wds.Binned2Norm(
(lambda ctx: ctx.model.layers[0].action.weight.data.detach()),
dig, wds_training_dir, 'Induced Changes in $W^{(1)}$'))
#.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True, layer_indices=plot_layers))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join())).reg(
tnr.CopyLogOnFinish(logpath)).reg(
tnr.ZipDirOnFinish(dtt_training_dir)).reg(
tnr.ZipDirOnFinish(pca_training_dir)).reg(
tnr.ZipDirOnFinish(pca3d_training_dir)).reg(
tnr.ZipDirOnFinish(pr_training_dir)).reg(
tnr.ZipDirOnFinish(svm_training_dir)).reg(
tnr.ZipDirOnFinish(satur_training_dir)).reg(
tnr.ZipDirOnFinish(trained_net_dir)))
trainer.train(network)
dig.archive_raw_inputs(os.path.join(SAVEDIR, 'digestor_raw.zip'))
def main():
"""Entry point"""
cu.DEFAULT_LINEAR_BIAS_INIT = wi.ZerosWeightInitializer()
cu.DEFAULT_LINEAR_WEIGHT_INIT = wi.GaussianWeightInitializer(
mean=0, vari=0.3, normalize_dim=0)
nets = cu.FluentShape(32 * 32 * 3).verbose()
network = FeedforwardComplex(INPUT_DIM, OUTPUT_DIM, [
nets.linear_(32 * 32 * 6),
nets.nonlin('isrlu'),
nets.linear_(500),
nets.nonlin('tanh'),
nets.linear_(250),
nets.nonlin('tanh'),
nets.linear_(250),
nets.nonlin('tanh'),
nets.linear_(100),
nets.tanh(),
nets.linear_(100),
nets.tanh(),
nets.linear_(100),
nets.tanh(),
nets.linear_(OUTPUT_DIM),
nets.nonlin('isrlu'),
])
train_pwl = CIFARData.load_train().to_pwl().restrict_to(set(
range(10))).rescale()
test_pwl = CIFARData.load_test().to_pwl().restrict_to(set(
range(10))).rescale()
layer_names = ('input', 'FC -> 32*32*6 (ISRLU)', 'FC -> 500 (tanh)',
'FC -> 250 (tang)', 'FC -> 250 (tanh)', 'FC -> 100 (tanh)',
'FC -> 100 (tanh)', 'FC -> 100 (tanh)',
f'FC -> {OUTPUT_DIM} (ISRLU)')
plot_layers = tuple(i for i in range(2, len(layer_names) - 1))
trainer = tnr.GenericTrainer(
train_pwl=train_pwl,
test_pwl=test_pwl,
teacher=FFTeacher(),
batch_size=45,
learning_rate=0.001,
optimizer=torch.optim.Adam(
[p for p in network.parameters() if p.requires_grad], lr=0.001),
criterion=torch.nn.CrossEntropyLoss())
pca3d_throughtrain.FRAMES_PER_TRAIN = 1
pca3d_throughtrain.SKIP_TRAINS = 16
pca3d_throughtrain.NUM_FRAME_WORKERS = 1
dig = npmp.NPDigestor('train_one_complex', 5)
dtt_training_dir = os.path.join(SAVEDIR, 'dtt')
pca_training_dir = os.path.join(SAVEDIR, 'pca')
pca3d_training_dir = os.path.join(SAVEDIR, 'pca3d')
pr_training_dir = os.path.join(SAVEDIR, 'pr')
svm_training_dir = os.path.join(SAVEDIR, 'svm')
satur_training_dir = os.path.join(SAVEDIR, 'saturation')
trained_net_dir = os.path.join(SAVEDIR, 'trained_model')
pca_throughtrain_dir = os.path.join(SAVEDIR, 'pca_throughtrain')
logpath = os.path.join(SAVEDIR, 'log.txt')
(trainer.reg(tnr.EpochsTracker()).reg(tnr.EpochsStopper(STOP_EPOCH)).reg(
tnr.DecayTracker()).reg(tnr.DecayStopper(8)).reg(
tnr.EpochProgress(print_every=120, hint_end_epoch=STOP_EPOCH)).reg(
tnr.LRMultiplicativeDecayer()).reg(
tnr.DecayOnPlateau(patience=3)).reg(
tnr.AccuracyTracker(1, 1000, True)).reg(
tnr.OnEpochCaller.create_every(
dtt.during_training_ff(dtt_training_dir, True,
dig),
skip=5)).reg(
tnr.OnEpochCaller.create_every(
pca_3d.during_training(
pca3d_training_dir,
True,
dig,
plot_kwargs={
'layer_names': layer_names
}),
start=10,
skip=100)).
reg(
tnr.OnEpochCaller.create_every(
pca_ff.during_training(pca_training_dir, True, dig), skip=5)).reg(
tnr.OnEpochCaller.create_every(
pr.during_training_ff(pr_training_dir,
True,
dig,
labels=False),
skip=5)).reg(
tnr.OnEpochCaller.create_every(
svm.during_training_ff(svm_training_dir, True,
dig),
skip=5)).reg(
tnr.OnEpochCaller.create_every(
satur.during_training(
satur_training_dir, True, dig),
skip=5)).reg(
tnr.OnEpochCaller.create_every(
tnr.save_model(trained_net_dir),
skip=5))
#.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True, layer_indices=plot_layers))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join())).reg(
tnr.ZipDirOnFinish(dtt_training_dir)).reg(
tnr.ZipDirOnFinish(pca_training_dir)).reg(
tnr.ZipDirOnFinish(pca3d_training_dir)).reg(
tnr.ZipDirOnFinish(pr_training_dir)).reg(
tnr.ZipDirOnFinish(svm_training_dir)).reg(
tnr.ZipDirOnFinish(satur_training_dir)).reg(
tnr.ZipDirOnFinish(trained_net_dir)).reg(
tnr.CopyLogOnFinish(logpath)))
trainer.train(network)
dig.archive_raw_inputs(os.path.join(SAVEDIR, 'digestor_raw.zip'))
def main():
"""Entry point"""
nets = cu.FluentShape(28*28)
network = FeedforwardComplex(
INPUT_DIM, OUTPUT_DIM,
[
nets.unflatten_conv_(1, 28, 28),
nets.conv_(5, 5, 5),
nets.relu(),
nets.maxpool_(2),
nets.flatten_(invokes_callback=True),
nets.linear_(nets.dims[0]),
nets.tanh(),
nets.linear_(OUTPUT_DIM),
nets.tanh()
]
)
#breakpoint()
train_pwl = MNISTData.load_train().to_pwl().restrict_to(set(range(10))).rescale()
test_pwl = MNISTData.load_test().to_pwl().restrict_to(set(range(10))).rescale()
layer_names = ('input', 'conv2d-relu', 'maxpool', 'tanh', 'output')
plot_layers = (3,)
trainer = tnr.GenericTrainer(
train_pwl=train_pwl,
test_pwl=test_pwl,
teacher=FFTeacher(),
batch_size=45,
learning_rate=0.001,
optimizer=torch.optim.Adam([p for p in network.parameters() if p.requires_grad], lr=0.001),
criterion=torch.nn.CrossEntropyLoss()
)
pca3d_throughtrain.FRAMES_PER_TRAIN = 1
pca3d_throughtrain.SKIP_TRAINS = 0
pca3d_throughtrain.NUM_FRAME_WORKERS = 6
dig = npmp.NPDigestor('train_one_complex', 35)
dtt_training_dir = os.path.join(SAVEDIR, 'dtt')
pca_training_dir = os.path.join(SAVEDIR, 'pca')
pca3d_training_dir = os.path.join(SAVEDIR, 'pca3d')
pr_training_dir = os.path.join(SAVEDIR, 'pr')
svm_training_dir = os.path.join(SAVEDIR, 'svm')
satur_training_dir = os.path.join(SAVEDIR, 'saturation')
trained_net_dir = os.path.join(SAVEDIR, 'trained_model')
pca_throughtrain_dir = os.path.join(SAVEDIR, 'pca_throughtrain')
(trainer
.reg(tnr.EpochsTracker())
.reg(tnr.EpochsStopper(5))
.reg(tnr.DecayTracker())
.reg(tnr.DecayStopper(8))
.reg(tnr.LRMultiplicativeDecayer())
.reg(tnr.DecayOnPlateau())
.reg(tnr.AccuracyTracker(5, 1000, True))
.reg(tnr.OnEpochCaller.create_every(dtt.during_training_ff(dtt_training_dir, True, dig), skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_3d.during_training(pca3d_training_dir, True, dig, plot_kwargs={'layer_names': layer_names}), skip=100))
#.reg(tnr.OnEpochCaller.create_every(pca_ff.during_training(pca_training_dir, True, dig), skip=100))
.reg(tnr.OnEpochCaller.create_every(pr.during_training_ff(pr_training_dir, True, dig), skip=100))
.reg(tnr.OnEpochCaller.create_every(svm.during_training_ff(svm_training_dir, True, dig), skip=100))
.reg(tnr.OnEpochCaller.create_every(satur.during_training(satur_training_dir, True, dig), skip=100))
.reg(tnr.OnEpochCaller.create_every(tnr.save_model(trained_net_dir), skip=100))
.reg(pca3d_throughtrain.PCAThroughTrain(pca_throughtrain_dir, layer_names, True, layer_indices=plot_layers))
.reg(tnr.OnFinishCaller(lambda *args, **kwargs: dig.join()))
.reg(tnr.ZipDirOnFinish(dtt_training_dir))
.reg(tnr.ZipDirOnFinish(pca_training_dir))
.reg(tnr.ZipDirOnFinish(pca3d_training_dir))
.reg(tnr.ZipDirOnFinish(pr_training_dir))
.reg(tnr.ZipDirOnFinish(svm_training_dir))
.reg(tnr.ZipDirOnFinish(satur_training_dir))
.reg(tnr.ZipDirOnFinish(trained_net_dir))
)
trainer.train(network)
dig.archive_raw_inputs(os.path.join(SAVEDIR, 'digestor_raw.zip'))