Skip to content

Instantly share code, notes, and snippets.

View jayleicn's full-sized avatar
🏂
snowboarding

Jie Lei 雷杰 jayleicn

🏂
snowboarding
View GitHub Profile
@jayleicn
jayleicn / mmf_trainer.py
Last active July 19, 2021 17:50
Logging other number in tensorbaord for MMF.
# in mmf/mmf/trainers/mmf_trainer.py
# add line below
from user import all_user_callbacks
# add two lines in
@registry.register_trainer("mmf")
class MMFTrainer(
TrainerCallbackHookMixin,
TrainerTrainingLoopMixin,
TrainerDeviceMixin,
TrainerEvaluationLoopMixin,
@jayleicn
jayleicn / bidaf.py
Created February 20, 2018 16:40
bidaf attention layer
import torch
import torch.nn as nn
class BidafAttnModule(nn.Module):
def __init__(self, hidden_size):
super(BidafAttnModule, self).__init__()
self.fc = nn.Sequential(
nn.Linear(hidden_size, hidden_size, bias=False),
nn.ReLU()
)
@jayleicn
jayleicn / forloop.sh
Last active January 5, 2018 06:10
start routine for pachong.rb
#!/bin/bash
mkdir -p download
mkdir -p fail
mkdir -p error
START=1
END=9500
STEP=500
rm -rf num_seq.txt
@jayleicn
jayleicn / mscoco_pascal
Created January 3, 2017 04:08
category mapping, MSCOCO and Pascal_VOC
MSCOCO Pascal_VOC
person person
bicycle bicycle
car car
motorcycle --motorbike
airplane --aeroplane
bus bus
train train
truck