Python Mask.get_mask_stats_by_idx Examples

Example #1

 def get_indices(tid, si):
     return set(
         Mask.get_mask_stats_by_idx(
             model.sess,
             mask_id,
             feed_dict={
                 X: model.trainXs[tid][si:si + batch_size_per_task],
                 is_training: True,
             })["indices"])

Example #2

File: mask_profile.py Project: dongkwan-kim/Selective-Forgetting-TF

def get_indices(tid, start_idx, batch_size, mask_ids, X, is_training):
    return set(Mask.get_mask_stats_by_idx(model.sess, mask_ids, feed_dict={
        X: model.trainXs[tid][start_idx:start_idx + batch_size],
        is_training: True,
    })["indices"])