summaryrefslogtreecommitdiff
path: root/models/model.py
AgeCommit message (Collapse)Author
2021-04-04Merge branch 'disentangling_only' into disentangling_only_py3.8disentangling_only_py3.8Jordan Gong
2021-04-04Add cross entropy lossdisentangling_onlyJordan Gong
2021-04-04Merge branch 'disentangling_only' into disentangling_only_py3.8Jordan Gong
# Conflicts: # models/model.py # utils/configuration.py # utils/sampler.py
2021-04-04Remove triplet samplerJordan Gong
2021-04-03Merge branch 'disentangling_only' into disentangling_only_py3.8Jordan Gong
# Conflicts: # models/model.py
2021-04-03Merge branch 'master' into disentangling_onlyJordan Gong
# Conflicts: # config.py # models/hpm.py # models/layers.py # models/model.py # models/part_net.py # models/rgb_part_net.py # test/part_net.py # utils/configuration.py # utils/triplet_loss.py
2021-04-03Revert "Normalize triplet losses"Jordan Gong
This reverts commit 99d1b18a
2021-03-27Normalize triplet lossesJordan Gong
2021-03-25Bug fixes and refactoringJordan Gong
1. Correct trained model signature 2. Move `val_size` to system config
2021-03-23Fix indexing bugs in validation dataset selectorJordan Gong
2021-03-22Add embedding visualization and validate on testing setJordan Gong
2021-03-16Set *_iter as *_iters in defaultJordan Gong
2021-03-15Remove redundant wrapper given by dataloaderJordan Gong
2021-03-15Fix redundant gallery_dataset_meta assignmentJordan Gong
2021-03-15Support transforming on training datasetsJordan Gong
2021-03-12Fix a typo when record none-zero countsJordan Gong
2021-03-12Make evaluate method staticJordan Gong
2021-03-12Code refactoringJordan Gong
1. Separate FCs and triplet losses for HPM and PartNet 2. Remove FC-equivalent 1x1 conv layers in HPM 3. Support adjustable learning rate schedulers
2021-03-10Bug fixesJordan Gong
1. Resolve reference problems when parsing dataset selectors 2. Transform gallery using different models
2021-03-04Replace detach with no_grad in evaluationJordan Gong
2021-03-04Set seed for reproducibilityJordan Gong
2021-03-02Record learning rate every stepJordan Gong
2021-03-02Fix bugs in new schedulerJordan Gong
2021-03-01New scheduler and new configJordan Gong
2021-03-01Move pairs variable to localJordan Gong
2021-02-28Implement sum of loss default in [1]Jordan Gong
[1]A. Hermans, L. Beyer, and B. Leibe, “In defense of the triplet loss for person re-identification,” arXiv preprint arXiv:1703.07737, 2017.
2021-02-28Log n-ile embedding distance and normJordan Gong
2021-02-27Implement Batch Hard triplet loss and soft marginJordan Gong
2021-02-26Fix predict functionJordan Gong
2021-02-20Separate triplet loss from modelJordan Gong
2021-02-19Merge branch 'python3.8' into disentangling_only_py3.8Jordan Gong
# Conflicts: # models/hpm.py # models/layers.py # models/model.py # models/part_net.py # models/rgb_part_net.py # utils/configuration.py
2021-02-19New branch with auto-encoder onlyJordan Gong
2021-02-19Merge branch 'master' into python3.8Jordan Gong
2021-02-19Allow evaluate unfinished modelJordan Gong
2021-02-18Merge branch 'master' into python3.8Jordan Gong
2021-02-18Implement adjustable input size and change some default configsJordan Gong
2021-02-18Decode mean appearance featureJordan Gong
2021-02-16Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py
2021-02-16Split transform and evaluate methodJordan Gong
2021-02-15Revert "Memory usage improvement"Jordan Gong
This reverts commit be508061
2021-02-15Revert "Memory usage improvement"Jordan Gong
This reverts commit be508061
2021-02-14Merge branch 'master' into python3.8Jordan Gong
2021-02-14Memory usage improvementJordan Gong
This update separates input data to two batches, which reduces ~30% memory usage.
2021-02-14Prepare for DataParallelJordan Gong
2021-02-10Merge branch 'master' into python3.8Jordan Gong
2021-02-10Save scheduler state_dictJordan Gong
2021-02-09Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/rgb_part_net.py
2021-02-09Improve performance when disentanglingJordan Gong
This is a HUGE performance optimization, up to 2x faster than before. Mainly because of the replacement of randomized for-loop with randomized tensor.
2021-02-09Some optimizationsJordan Gong
1. Scheduler will decay the learning rate of auto-encoder only 2. Write learning rate history to tensorboard 3. Reduce image log frequency
2021-02-08Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/hpm.py # models/layers.py # models/model.py # models/rgb_part_net.py # utils/configuration.py
generated by cgit v1.2.3 (git 2.25.1) at 2024-12-23 18:14:15 +0000