fork_resnet_remove_distillation.m

function [net, derOutputs] = fork_resnet_remove_distillation(net, derOutputs)

%% Update loss layer for the old layer. Only the "last new" task is updated.
index = strfind({net.layers.name}, 'loss_distillation');
index = find(not(cellfun('isempty', index)));
    
net.removeLayer(net.layers(index(1)).name);
derOutputs = derOutputs(1:end-2);